import sys
ZADANIE = sys.stdin.read().strip().splitlines()
ENE, DUE, RIKE = [list(map(int, napis)) for napis in ZADANIE]
FAKE = [e+d-r for e,d,r in zip(ENE, DUE, RIKE)]
BORBA = False
OSMESMAKE = []
for _,KORBA in enumerate(FAKE):
if not BORBA:
if KORBA == -1:
BORBA = True
else:
OSMESMAKE.append(KORBA)
else:
if KORBA == -1:
OSMESMAKE.append(-1)
elif KORBA == 0:
OSMESMAKE.append(-1)
OSMESMAKE.append(0)
BORBA = False
elif KORBA == 10:
OSMESMAKE.append(0)
BORBA = False
elif KORBA == 9:
pass
else:
BORBA = False
OSMESMAKE.append(KORBA)
DEUS = 0
DEUSKOSMADEUS = []
for _, val in enumerate(OSMESMAKE):
if val == 0:
DEUS += 1
else:
if DEUS > 0:
DEUSKOSMADEUS.append(int(DEUS*(DEUS+1)/2))
DEUS = 0
DEUSKOSMADEUS.append(int(DEUS*(DEUS+1)/2))
# i...
MORELE_BEC = sum(DEUSKOSMADEUS)
print(MORELE_BEC)
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 | import sys ZADANIE = sys.stdin.read().strip().splitlines() ENE, DUE, RIKE = [list(map(int, napis)) for napis in ZADANIE] FAKE = [e+d-r for e,d,r in zip(ENE, DUE, RIKE)] BORBA = False OSMESMAKE = [] for _,KORBA in enumerate(FAKE): if not BORBA: if KORBA == -1: BORBA = True else: OSMESMAKE.append(KORBA) else: if KORBA == -1: OSMESMAKE.append(-1) elif KORBA == 0: OSMESMAKE.append(-1) OSMESMAKE.append(0) BORBA = False elif KORBA == 10: OSMESMAKE.append(0) BORBA = False elif KORBA == 9: pass else: BORBA = False OSMESMAKE.append(KORBA) DEUS = 0 DEUSKOSMADEUS = [] for _, val in enumerate(OSMESMAKE): if val == 0: DEUS += 1 else: if DEUS > 0: DEUSKOSMADEUS.append(int(DEUS*(DEUS+1)/2)) DEUS = 0 DEUSKOSMADEUS.append(int(DEUS*(DEUS+1)/2)) # i... MORELE_BEC = sum(DEUSKOSMADEUS) print(MORELE_BEC) |
English