Update day 2 solution

Don't split into list
This commit is contained in:
2021-12-02 18:25:26 +01:00
parent 3700b24968
commit e5f751fd0f

View File

@@ -4,13 +4,13 @@ def part_1(input):
h_pos = 0
depth = 0
for line in input:
step = line.split()
if 'forward' == step[0]:
h_pos += int(step[1])
elif 'up' == step[0]:
depth -= int(step[1])
elif 'down' == step[0]:
depth += int(step[1])
cmd, rng = line.split()
if 'forward' == cmd:
h_pos += int(rng)
elif 'up' == cmd:
depth -= int(rng)
elif 'down' == cmd:
depth += int(rng)
result = h_pos * depth
print("Part 1 result:", result)
@@ -20,14 +20,14 @@ def part_2(input):
depth = 0
aim = 0
for line in input:
step = line.split()
if 'forward' == step[0]:
h_pos += int(step[1])
depth += aim * int(step[1])
elif 'up' == step[0]:
aim -= int(step[1])
elif 'down' == step[0]:
aim += int(step[1])
cmd, rng = line.split()
if 'forward' == cmd:
h_pos += int(rng)
depth += aim * int(rng)
elif 'up' == cmd:
aim -= int(rng)
elif 'down' == cmd:
aim += int(rng)
result = h_pos * depth
print("Part 2 result:", result)