32 lines
736 B
Python
Executable File
32 lines
736 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"):
|
|
|
|
l = []
|
|
for item in line:
|
|
i = int(item)
|
|
l.append(i)
|
|
|
|
for i in range(len(l)):
|
|
for i2 in range(i + 1, len(l)):
|
|
a1 = l[i] / l[i2]
|
|
a2 = l[i2] / l[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))
|