blob: 0dc174a2773527cc993962c1985105389503f43b (
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
42
43
44
45
46
47
48
49
50
51
52
|
#advent of code 2018
#day 08
#part 1 and 2
f = open("08.in", 'r');
myinput = eval(f.read().replace(" ",","));
f.close();
#part 1
def NodeAnalysis_1(License, index):
NumberOfChilds = License[index[0]];
index[0] += 1;
NumberofEntries = License[index[0]];
index[0] += 1;
for child in range(NumberOfChilds):
NodeAnalysis_1(License, index);
for entry in range(NumberofEntries):
index[1] += License[index[0]];
index[0] += 1;
#part 2
def NodeAnalysis_2(License, index):
NodeVal = 0;
NumberOfChilds = License[index[0]];
index[0] += 1;
NumberofEntries = License[index[0]];
index[0] += 1;
ListOfEntries = [];
ListOfChilds = [];
for child in range(NumberOfChilds):
ListOfChilds.append(NodeAnalysis_2(License, index));
for entry in range(NumberofEntries):
ListOfEntries.append(License[index[0]]);
index[0] += 1;
if(NumberOfChilds == 0):
NodeVal = sum(ListOfEntries);
else:
for e in ListOfEntries:
if (e == 0 or e >NumberOfChilds):
continue;
NodeVal+= ListOfChilds[e-1];
index.append(NodeVal);
return NodeVal;
i = [0,0];
NodeAnalysis_1(myinput, i);
print("part 1 =", i[-1]);
part2 = NodeAnalysis_2(myinput, [0,0]);
print("part 2 =", part2);
|
