Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- with open('filename', 'r') as f:
- n = int(f.readline())
- nums = []
- for i in range(n):
- nums.append(int(f.readline()))
- l1 = [[0, 0, 0] for i in range(n)]
- for i in range(n - 1, -1, -1):
- if i == n - 1:
- l2 = [0, 0, 0]
- else:
- l2 = l1[i + 1][:]
- if nums[i] % 14 == 0:
- l2[0] += 1
- elif nums[i] % 2 == 0:
- l2[1] += 1
- elif nums[i] % 7 == 0:
- l2[2] += 1
- l1[i] = l2
- res = 0
- for i in range(n):
- if i + 7 >= n:
- break
- if nums[i] % 14 == 0:
- res += n - (i + 7)
- elif nums[i] % 2 == 0:
- res += l1[i + 7][2] + l1[i + 7][0]
- elif nums[i] % 7 == 0:
- res += l1[i + 7][1] + l1[i + 7][0]
- else:
- res += l1[i + 7][0]
- print(res)
Add Comment
Please, Sign In to add comment