day2.py

input = open('input/day2.txt', 'r').readlines()
directions = [line for line in input]

# O(n)
def partOne(directions=directions):
    horizontal = 0
    depth = 0

    for step in directions:
        direction, val = step.split(" ", 1)
        if direction == 'forward':
            horizontal += int(val)
        elif direction == 'down':
            depth += int(val)
        elif direction == 'up':
            depth -= int(val)
    
    return horizontal*depth

# # O(n)
def partTwo(directions=directions):
    horizontal = 0
    depth = 0
    aim = 0

    for step in directions:
        direction, val = step.split(" ", 1)
        if direction == 'forward':
            horizontal += int(val)
            depth += aim*int(val)
        elif direction == 'down':
            aim += int(val)
        elif direction == 'up':
            aim -= int(val)
    
    return horizontal*depth

print("Part One:", partOne())
print("Part Two:", partTwo())