39 lines
979 B
Python
Executable file
39 lines
979 B
Python
Executable file
#!/usr/bin/env python
|
|
|
|
import sys
|
|
from collections import Counter
|
|
|
|
def most_common(ns, pos):
|
|
c = Counter(n[pos] for n in ns)
|
|
num_zeros = c['0']
|
|
num_ones = c['1']
|
|
if num_ones >= num_zeros:
|
|
return '1'
|
|
else:
|
|
return '0'
|
|
|
|
def least_common(ns, pos):
|
|
return '0' if most_common(ns, pos) == '1' else '1'
|
|
|
|
def find_criteria(ns, criteria):
|
|
ns = ns
|
|
i = 0
|
|
while len(ns) > 1:
|
|
x = criteria(ns, i)
|
|
ns = [n for n in ns if n[i] == x]
|
|
i += 1
|
|
return ns[0]
|
|
|
|
if __name__ == '__main__':
|
|
lines = [line.strip() for line in sys.stdin.readlines()]
|
|
|
|
line_len = len(lines[0])
|
|
gamma = ''.join(most_common(lines, n) for n in range(line_len))
|
|
epsilon = ''.join(least_common(lines, n) for n in range(line_len))
|
|
print(int(gamma, 2) * int(epsilon, 2))
|
|
|
|
oxygen_rating = find_criteria(lines, most_common)
|
|
co2_rating = find_criteria(lines, least_common)
|
|
|
|
print(int(oxygen_rating, 2) * int(co2_rating, 2))
|