blob: 7fb7c08add7eef946ad1003afc316cb515b320cf (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
#!/usr/bin/env python3
from functools import reduce
from re import findall
from copy import deepcopy
import sys
# import numpy as np
filename = sys.argv[1] if len(sys.argv) == 2 \
else "in/" + sys.argv[0].split('/')[-1].rstrip(".py") + ".pzl"
data = open(filename).read()
blocks = data.rstrip('\n').split('\n\n')
res1 = 0
res2 = 0
S = []
for i, block in enumerate(blocks[:-1]):
s = 0
for line in block.split('\n')[1:]:
for c in line:
if c == '#':
s += 1
S.append(s)
print(S)
for line in blocks[-1].split('\n'):
l, r = line.split(':')
x, y = l.split('x')
x, y = int(x), int(y)
sh = [int(i) for i in r.strip().split(' ')]
a = sum([i*j for i, j in zip(sh, S)])
print(a / (x*y))
print('res1:', res1)
print('res2:', res2)
|
