49 lines
918 B
Python
49 lines
918 B
Python
import sys
|
|
|
|
# PART 1
|
|
|
|
def load_buckets(filename):
|
|
with open(filename, "r") as f:
|
|
return tuple(map(int, f.read()[:-1].split("\t")))
|
|
|
|
def redistribute(buckets):
|
|
l = list(buckets)
|
|
i = l.index(max(l))
|
|
n = l[i]
|
|
l[i] = 0
|
|
|
|
while n > 0:
|
|
i = (i + 1) % len(l)
|
|
l[i] += 1
|
|
n -= 1
|
|
|
|
return tuple(l)
|
|
|
|
def find_repeat(buckets):
|
|
cycles = 0
|
|
states = {buckets: 0}
|
|
while True:
|
|
buckets = redistribute(buckets)
|
|
cycles += 1
|
|
if buckets in states:
|
|
return cycles, states[buckets]
|
|
else:
|
|
states[buckets] = cycles
|
|
|
|
# PART 2
|
|
|
|
def loop_cycles(buckets):
|
|
cycles, start_cycle = find_repeat(buckets)
|
|
return cycles - start_cycle
|
|
|
|
def main(filename):
|
|
print(f"Solutions for {filename}")
|
|
buckets = load_buckets(filename)
|
|
cycles, _ = find_repeat(buckets)
|
|
print(f"Part 1: {cycles}")
|
|
cycles_2 = loop_cycles(buckets)
|
|
print(f"Part 2: {cycles_2}")
|
|
|
|
if __name__ == "__main__":
|
|
for filename in sys.argv[1:]:
|
|
main(filename)
|