blob: c9707ab3e1c62541b53595fd076936105e901649 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
import fileinput
from functools import reduce
from operator import mul
a = []
op = []
with fileinput.input() as lines:
for line in lines:
t = line[:-1].split(' ')
if t[0] == '+' or t[0] == '*':
op = [x for x in t if x != '']
else:
a.append([int(x) for x in t if x != ''])
def sol(a, op, i):
col = [a[j][i] for j in range(len(a))]
return sum(col) if op[i] == '+' else reduce(mul, col, 1)
print(sum(sol(a, op, i) for i in range(len(a[0]))))
|
