Advent of Code Day 2

day2_pt1.py

def run_program(data):
    for i in range(0, len(data), 4):
        if i == 0 or i%4 == 0:
            if data[i] == 99:
                return data[0]
            elif data[i] == 1:
                data[data[i+3]] = data[data[i+1]] + data[data[i+2]]
            elif data[i] == 2:
                data[data[i+3]] = data[data[i+1]] * data[data[i+2]]
            else:
                print("---------------------------FOUND BAD INSTRUCTION CODE")
                return -1
    return data[0]

## Load Data
with open("input.txt", "r") as f:
    data = f.readlines()

## Format Data, Set Variables
data = data[0].split(",")
data = [int(x) for x in data]
data[1] = 12
data[2] = 2

## Run Program
print("Day2, Part1 --", run_program(data))

day2_pt2.py

def run_program(data):
    for i in range(0, len(data), 4):
        if i == 0 or i%4 == 0:
            if data[i] == 99:
                return data[0]
            elif data[i] == 1:
                data[data[i+3]] = data[data[i+1]] + data[data[i+2]]
            elif data[i] == 2:
                data[data[i+3]] = data[data[i+1]] * data[data[i+2]]
            else:
                print("---------------------------FOUND BAD INSTRUCTION CODE")
                return -1
    return data[0]

def find_corret_inputs(data, pointer_code):
    for first_i in range(0, 99):
        for second_i in range(0, 99):
            new_data = list(data)
            new_data[1], new_data[2] = first_i, second_i
            if run_program(new_data) == pointer_code:
                return (100 * first_i + second_i)
            else:
                del(new_data)

## Load Data
with open("input.txt", "r") as f:
    data = f.readlines()

## Format Data
data = data[0].split(",")
data = [int(x) for x in data]
pointer_code = 19690720

## Run Program
print("Day2, Part2 --", find_corret_inputs(data, pointer_code))