Advent of Code 2024

input_str = open('input1.txt').read().strip()

# part one
list_one, list_two = zip(*(map(int, line.split("   ")) 
    for line in input_str.split("\n"))
)


s = sum(
    abs(a - b) for a, b in zip(
        sorted(list_one), 
        sorted(list_two))
)
print(s)

# part two

import collections
counter = collections.Counter(list_two)
similarity = sum([x * counter[x] for x in list_one if x in counter])
print(similarity)

input_str = open('input2.txt').read().splitlines()

# part one
is_safe = lambda levels: all(
    (1 <= abs(a - b) <= 3) and ((a - b >= 0) == (levels[0] - levels[1] >= 0))
    for a, b in zip(levels, levels[1:])
)


safe_count = sum(is_safe([int(x) for x in line.split()]) for line in input_str)    
print(safe_count)

# part two
def safe_count_part_2():
    s_c = 0
    for report in input_str:
        levels = [int(x) for x in report.split()]
        if is_safe(levels):
            s_c += 1
        else:
            for i in range(len(levels)):
                if is_safe(levels[:i] + levels[i + 1 :]):
                    s_c += 1
    return s_c

print(safe_count_part_2)

import re
input_str = open("input3.txt").read()

# `part one`

multiplications_instructions_re = re.compile(r"mul\(\w+,\w+\)")

count = sum([eval(input_str[i.span()[0]+4:i.span()[1]-1].replace(",", "*")) for i in multiplications_instructions_re.finditer(input_str)])
print(count)


# part two

# don't judge my regex. very pro - i think we have thing like group but meeeh
token_instructions_re = re.compile(r"mul\(\w+,\w+\)|do\(\)|don't\(\)")
program: list[str] = token_instructions_re.findall(input_str)

enabled = True

total = 0

for ins in program:
    if 'mul' in ins and enabled:
        total += eval(ins[4:-1].replace(",", "*"))
    elif ins == 'do()':
        enabled = True
    elif ins == "don't()":
        enabled = False
print(total)