aoc2015/d06/part2.py

19 lines
493 B
Python

import sys
M = [[0 for _ in range(1000)] for _ in range(1000)]
for l in sys.stdin.readlines():
I = l[:-1].split(' ')
xf,yf = list(map(int,I[-3].split(',')))
xt,yt = list(map(int,I[-1].split(',')))
for x in range(xf,xt+1):
for y in range(yf,yt+1):
if I[-4] == 'on':
M[x][y] += 1
elif I[-4] == 'off':
M[x][y] = max(0, M[x][y]-1)
else:
M[x][y] += 2
print(sum(x for C in M for x in C))