H0_0H

Untitled

Dec 17th, 2021 (edited)
104
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.88 KB | None | 0 0
  1. with open('filename', 'r') as f:
  2.     n = int(f.readline())
  3.     nums = []
  4.     for i in range(n):
  5.         nums.append(int(f.readline()))
  6.    
  7.     l1 = [[0, 0, 0] for i in range(n)]
  8.  
  9.     for i in range(n - 1, -1, -1):
  10.         if i == n - 1:
  11.             l2 = [0, 0, 0]
  12.         else:
  13.             l2 = l1[i + 1][:]
  14.        
  15.         if nums[i] % 14 == 0:
  16.             l2[0] += 1
  17.         elif nums[i] % 2 == 0:
  18.             l2[1] += 1
  19.         elif nums[i] % 7 == 0:
  20.             l2[2] += 1
  21.  
  22.         l1[i] = l2
  23.    
  24.     res = 0
  25.     for i in range(n):
  26.         if i + 7 >= n:
  27.             break
  28.         if nums[i] % 14 == 0:
  29.             res += n - (i + 7)
  30.         elif nums[i] % 2 == 0:
  31.             res += l1[i + 7][2] + l1[i + 7][0]
  32.         elif nums[i] % 7 == 0:
  33.             res += l1[i + 7][1] + l1[i + 7][0]
  34.         else:
  35.             res += l1[i + 7][0]
  36.            
  37.     print(res)
Add Comment
Please, Sign In to add comment