first commit

06/part2.py

@@ -0,0 +1,69 @@
+#!/usr/bin/python
+
+import re
+
+
+def read_file(filename):
+    file = open(filename, 'r')
+    while True:
+        line = file.readline()
+        if not line:
+            break
+        yield line
+
+
+def getPosKey(posX, posY):
+    return "%03d%03d" % (posX, posY)
+
+
+def action_turn_on(data, posX1, posY1, posX2, posY2):
+    changed = 0
+    for posX in range(posX1, posX2+1):
+        for posY in range(posY1, posY2+1):
+            key = getPosKey(posX, posY)
+            if key not in data:
+                data[key] = 0
+            data[key] = data[key] + 1
+            changed = changed + 1
+    return (data, changed, 0)
+
+
+def action_turn_off(data, posX1, posY1, posX2, posY2):
+    changed = 0
+    for posX in range(posX1, posX2+1):
+        for posY in range(posY1, posY2+1):
+            key = getPosKey(posX, posY)
+            if key not in data:
+                data[key] = 0
+            if data[key] > 0:
+                data[key] = data[key] - 1
+                changed = changed + 1
+    return (data, 0, changed)
+
+def main():
+    RE_COORD = re.compile(r" (\d+),(\d+) through (\d+),(\d+)$")
+    lOn = 0
+    grid = dict()
+    for line in read_file('input'):
+        match = re.search(RE_COORD, line)
+        posX1 = (int)(match.group(1))
+        posY1 = (int)(match.group(2))
+        posX2 = (int)(match.group(3))
+        posY2 = (int)(match.group(4))
+        changedOn = 0
+        changedOff = 0
+        if line.startswith("turn on "):
+            (grid, changedOn, changedOff) = action_turn_on(grid, posX1, posY1, posX2, posY2)
+        if line.startswith("turn off "):
+            (grid, changedOn, changedOff) = action_turn_off(grid, posX1, posY1, posX2, posY2)
+        if line.startswith("toggle "):
+            (grid, changedOn, changedOff) = action_turn_on(grid, posX1, posY1, posX2, posY2)
+            (grid, changedOn, changedOff) = action_turn_on(grid, posX1, posY1, posX2, posY2)
+            changedOn = changedOn * 2
+            changedOff = changedOff * 2
+        lOn = lOn + changedOn - changedOff
+
+    print lOn
+
+if __name__ == "__main__":
+    main()