結果

問題 No.1240 Or Sum of Xor Pair
コンテスト
ユーザー onakasuitacity
提出日時 2020-10-24 20:39:11
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 847 bytes
コンパイル時間 340 ms
コンパイル使用メモリ 82,248 KB
実行使用メモリ 260,928 KB
最終ジャッジ日時 2024-07-21 15:40:11
合計ジャッジ時間 36,343 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 3
other WA * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

def _fourier(f, inverse = False):
    f = f[:]
    n = (len(f) - 1).bit_length()
    for d in range(n):
        for U in range(1 << (n - 1)):
            if not U >> d & 1:
                s, t = f[U], f[U | 1 << d]
                f[U], f[U | 1 << d] = s + t, s - t
    if inverse:
        f = [v >> n for v in f]
    return f

def convolution(f, g):
    return _fourier([a * b for a, b in zip(_fourier(f), _fourier(g))], inverse = 1)

n, x = map(int, input().split())
A = list(map(int, input().split()))
m = max(A).bit_length()

f = [0] * (1 << m)
for a in A:
    f[a] += 1
f2 = convolution(f, f)
k = (sum(f2[:x]) - sum(f)) // 2

ans = 0
for d in range(m):
    f = [0] * (1 << m)
    for a in A:
        if not a >> d & 1:
            f[a] += 1

    f2  = convolution(f, f)
    l = (sum(f2[:x]) - sum(f)) // 2
    ans += (k - l) << d

print(ans)
0