blob: 552d5b9ea15c0b382fe38af000231c36da522f35 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
#!/usr/bin/env python3
from functools import reduce
from re import findall
from copy import deepcopy
import sys
# import numpy as np
filename = sys.argv[1] if len(sys.argv) == 2 \
else "in/" + sys.argv[0].split('/')[-1].rstrip(".py") + ".pzl"
data = open(filename).read()
lines = data.rstrip('\n').split('\n')
res1 = 0
res2 = 0
def check1(a):
a = str(a)
l = len(a)
if l % 2 != 0:
return False
if a[0:l//2] == a[l//2:]:
return True
return False
def check2(a):
a = str(a)
l = len(a)
for n in range(1, l):
aa = [a[i:i+n] for i in range(0, l, n)]
if len(a) > 1 and len(set(aa)) == 1:
return True
return False
line = lines[0]
for seq in line.split(','):
a, b = seq.split('-')
a, b = int(a), int(b)
print(b - a)
for i in range(a, b+1):
if check1(i):
res1 += i
if check2(i):
res2 += i
print('res1:', res1)
print('res2:', res2)
|
