27.07 Прочие прототипы

from math import comb

f = open(’1.txt’)
n = int(f.readline())
ans = 0
k = [0, 0, 0]

for i in range(n):
    x = int(f.readline())
    k[x % 3] += 1

s2 = [0, 0, 0]
for i in range(3):
    for j in range(i, k[2] + 1, 3):
        s2[i] += comb(k[2], j)

s1 = [0, 0, 0]
for i in range(3):
    for j in range(i, k[1] + 1, 3):
        s1[i] += comb(k[1], j)

ans = s1[0] * s2[0] + s1[1] * s2[1] + s1[2] * s2[2]
ans = ans * 2 ** k[0]

print(ans)