Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
Nice and simple solution in Clear category for The Square Chest by quiath
from collections import defaultdict
N = 4
def checkio(lines_list):
"""Return the quantity of squares"""
c = 0
l, u = defaultdict(int), defaultdict(int)
lines_list = sorted([ sorted(_) for _ in lines_list ])
for p in lines_list:
if p[1] == p[0] + 1:
l[p[1]] = l[p[0]] + 1
elif p[1] == p[0] + N:
u[p[1]] = u[p[0]] + 1
for k in l:
m = min(l[k], u[k])
for i in range(1, m + 1):
if u.get(k - i, 0) >= i <= l.get(k - i * N, 0):
c += 1
return c
if __name__ == '__main__':
assert (checkio([[1, 2], [3, 4], [1, 5], [2, 6], [4, 8], [5, 6], [6, 7],
[7, 8], [6, 10], [7, 11], [8, 12], [10, 11],
[10, 14], [12, 16], [14, 15], [15, 16]]) == 3), "First, from description"
assert (checkio([[1, 2], [2, 3], [3, 4], [1, 5], [4, 8],
[6, 7], [5, 9], [6, 10], [7, 11], [8, 12],
[9, 13], [10, 11], [12, 16], [13, 14], [14, 15], [15, 16]]) == 2), "Second, from description"
assert (checkio([[1, 2], [1, 5], [2, 6], [5, 6]]) == 1), "Third, one small square"
assert (checkio([[1, 2], [1, 5], [2, 6], [5, 9], [6, 10], [9, 10]]) == 0), "Fourth, it's not square"
assert (checkio([[16, 15], [16, 12], [15, 11], [11, 10],
[10, 14], [14, 13], [13, 9]]) == 0), "Fifth, snake"
Oct. 2, 2014
