結果

問題 No.1240 Or Sum of Xor Pair
コンテスト
ユーザー onakasuitacity
提出日時 2020-10-24 20:10:51
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,335 ms / 2,000 ms
コード長 1,086 bytes
コンパイル時間 589 ms
コンパイル使用メモリ 81,988 KB
実行使用メモリ 233,076 KB
最終ジャッジ日時 2024-07-21 15:35:24
合計ジャッジ時間 37,573 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
INF = 1 << 60
MOD = 10**9 + 7 # 998244353
sys.setrecursionlimit(2147483647)
input = lambda:sys.stdin.readline().rstrip()

def fourier(f, inverse = False):
    n = (len(f) - 1).bit_length()
    for d in range(n):
        for U in range(1 << n):
            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 resolve():
    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
    Ff = fourier(f)
    Ff2 = [v * v for v in Ff]
    f2 = fourier(Ff2, inverse = True)
    k = (sum(f2[:x]) - n) // 2

    ans = 0
    for d in range(m):
        f = [0] * (1 << m)
        for a in A:
            if not a >> d & 1:
                f[a] += 1
        l = sum(f)
        Ff = fourier(f)
        Ff2 = [v * v for v in Ff]
        f2 = fourier(Ff2, inverse = True)
        l = (sum(f2[:x]) - l) // 2
        ans += (k - l) << d

    print(ans)
resolve()
0