33 lines
767 B
Python
Executable File
33 lines
767 B
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import csv
|
|
|
|
inputs = ["part2.txt", "input.txt"]
|
|
|
|
|
|
def solve(filename):
|
|
total = 0
|
|
with open(filename) as tsv:
|
|
for line in csv.reader(tsv, dialect="excel-tab"):
|
|
|
|
items = []
|
|
for item in line:
|
|
i = int(item)
|
|
items.append(i)
|
|
|
|
for i in range(len(items)):
|
|
for i2 in range(i + 1, len(items)):
|
|
a1 = items[i] / items[i2]
|
|
a2 = items[i2] / items[i]
|
|
if float.is_integer(a1):
|
|
total += a1
|
|
elif float.is_integer(a2):
|
|
total += a2
|
|
|
|
return int(total)
|
|
|
|
|
|
for puzzle in inputs:
|
|
print(puzzle, solve(puzzle))
|