-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path9.py
45 lines (40 loc) · 1.07 KB
/
9.py
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
42
43
44
45
import sys
import functools
from lib import (
read_file
)
if len(sys.argv) > 1:
test_run = int(sys.argv[1])
else:
test_run = 1
if test_run == 1:
arr = read_file("exem")
else:
arr = read_file("input")
part1 = False
sum = 0
if part1:
for line in arr:
line = [int(i) for i in line.split(" ")]
iter = len(line)
while iter > 0 and line[:iter] != [0 for i in range(iter)]:
iter -= 1
for i in range(iter):
line[i] = line[i+1]-line[i]
for i in range(len(line)-1):
if line[i] != 0:
line[i+1] += line[i]
sum += line[-1]
else:
for line in arr:
line = [int(i) for i in line.split(" ")]
iter = 0
while iter < len(line) and line[iter:] != [0 for i in range(len(line)-iter)]:
for i in range(len(line)-1,iter,-1):
line[i] = -(line[i-1]-line[i])
iter += 1
for i in range(len(line)-1,-1,-1):
if line[i] != 0:
line[i-1] -= line[i]
sum += line[0]
print(sum)