Advertisement
Pastehsjsjs

Untitled

Apr 10th, 2023
578
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.73 KB | Source Code | 0 0
  1. import math
  2.  
  3.  
  4. def get_n(x):
  5.     n = math.log(x, 2)
  6.     if float(n) != int(n) or n == 0:
  7.         return 0
  8.     n = int(n)
  9.     if n >= 10:
  10.         return 10
  11.     return n
  12.  
  13.  
  14. with open('/Users/alex26/Desktop/27/27-A.txt', 'r') as f:
  15.     data = [[0 for j in range(1024)] for _ in range(3)]
  16.     res = 0
  17.     for i in range(int(f.readline())):
  18.         x = int(f.readline())
  19.         n = get_n(x)
  20.         nap = (3 - (x % 3)) % 3
  21.  
  22.         if n == 0:
  23.             res += data[nap][0]
  24.         elif n != 10:
  25.             for s in range((10 - n), 10 + 1):
  26.                 res += data[nap][2 ** s - 1]
  27.         elif n == 10:
  28.             for val in data[nap]:
  29.                 res += val
  30.  
  31.         data[x % 3][x % 1024] += 1
  32.  
  33. print(res)
  34.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement