Files
aocpy/2017/d1.py

15 lines
272 B
Python

data = open(0).read().strip()
s = 0
for i in range(len(data)):
if data[i] == data[(i + 1) % len(data)]:
s += int(data[i])
print(s)
s = 0
for i in range(len(data)):
if data[i] == data[(i + len(data) // 2) % len(data)]:
s += int(data[i])
print(s)