Merge pull request #26 from sqdorte/master

calypso · web-flow · commit d0536f66598a · 2018-12-03T17:49:28.000-02:00
03/2 - python
diff --git a/2018/03_2/python/sqdorte.py b/2018/03_2/python/sqdorte.py
@@ -0,0 +1,36 @@
+with open('input', 'r') as raw:
+    i = raw.read().strip().split('\n')
+
+claimed = {}
+claims = []
+
+for x in i:
+    claim = x.split('@ ')
+    
+    cid = claim[0][1:]
+    coords = claim[1].split(': ')
+
+    a,b = map(int, coords[0].split(','))
+    c,d = map(int, coords[1].split('x'))
+
+    for x in range(a, a+c):
+        for y in range(b, b+d):
+            try:
+                claimed[(x,y)] += 1
+            except KeyError:
+                claimed[(x,y)] = 1
+
+    claims.append((cid,a,b,c,d))
+
+for cid,a,b,c,d in claims:
+    s = True
+    for x in range(a, a+c):
+        if not s:
+            break
+        for y in range(b, b+d):
+            if claimed[(x,y)] > 1:
+                s = False
+                break
+    if s:
+        print(cid)
+        break
