Jake Pullen
75 lines
2.9 KiB
Python
75 lines
2.9 KiB
Python
with open(r'advent_of_code\2023\15\input.txt', 'r') as file:
|
|
input_data = file.read()
|
|
|
|
test_input = 'rn=1,cm-,qp=3,cm=2,qp-,pc=4,ot=9,ab=5,pc-,pc=6,ot=7'
|
|
# input_data = test_input
|
|
|
|
# Function to calculate hash value
|
|
def calculate_hash(string):
|
|
hash_value = 0
|
|
for char in string:
|
|
hash_value = ((hash_value + ord(char)) * 17) % 256
|
|
return hash_value
|
|
|
|
# Function to process commands
|
|
def process_commands(commands, box):
|
|
# Iterate over each command in the list of commands
|
|
for command in commands:
|
|
# Check if the last character of the command is '-'
|
|
if command[-1] == '-':
|
|
# If it is, the command is to remove a name from the box
|
|
# Extract the name from the command by removing the last character
|
|
name_to_remove = command[:-1]
|
|
# Calculate the hash value for the name
|
|
hash_value = calculate_hash(name_to_remove)
|
|
# Create a new list for the box at the hash value index, excluding the name to remove
|
|
box[hash_value] = [(name, value) for (name, value) in box[hash_value] if name != name_to_remove]
|
|
elif command[-2] == '=':
|
|
# If the second last character of the command is '=', the command is to add or update a name in the box
|
|
# Extract the name and the length from the command
|
|
name_to_add_or_update = command[:-2]
|
|
length = int(command[-1])
|
|
# Calculate the hash value for the name
|
|
hash_value = calculate_hash(name_to_add_or_update)
|
|
# Check if the name is already in the box at the hash value index
|
|
if name_to_add_or_update in [name for (name, value) in box[hash_value]]:
|
|
# If it is, update the length for the name
|
|
box[hash_value] = [(name, length if name == name_to_add_or_update else value) for (name, value) in box[hash_value]]
|
|
else:
|
|
# If it's not, add the name and length to the box at the hash value index
|
|
box[hash_value].append((name_to_add_or_update, length))
|
|
|
|
# Split the commands
|
|
commands = input_data.split(',')
|
|
|
|
# Calculate the sum of hash values
|
|
sum_hash_values = 0
|
|
for command in commands:
|
|
sum_hash_values += calculate_hash(command)
|
|
print(sum_hash_values)
|
|
|
|
# Initialize the box
|
|
box = [[] for _ in range(256)]
|
|
|
|
# Process the commands
|
|
process_commands(commands, box)
|
|
|
|
# Function to calculate the product of indices and value
|
|
def calculate_product(index1, index2, value):
|
|
return (index1 + 1) * (index2 + 1) * value
|
|
|
|
# Initialize the total sum
|
|
total_sum = 0
|
|
|
|
# Iterate over each box with its index
|
|
for box_index, box in enumerate(box):
|
|
# Iterate over each tuple in the box with its index
|
|
for tuple_index, (name, value) in enumerate(box):
|
|
# Calculate the product using the function
|
|
product = calculate_product(box_index, tuple_index, value)
|
|
|
|
# Add the product to the total sum
|
|
total_sum += product
|
|
|
|
# Print the total sum
|
|
print(total_sum) |