結果

問題 No.1240 Or Sum of Xor Pair
ユーザー onakasuitacityonakasuitacity
提出日時 2020-10-24 20:37:15
言語 PyPy3
(7.3.15)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,071 bytes
コンパイル時間 332 ms
コンパイル使用メモリ 82,216 KB
実行使用メモリ 260,772 KB
最終ジャッジ日時 2024-07-21 15:39:27
合計ジャッジ時間 59,139 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
53,328 KB
testcase_01 AC 44 ms
59,656 KB
testcase_02 AC 49 ms
63,672 KB
testcase_03 TLE -
testcase_04 TLE -
testcase_05 TLE -
testcase_06 TLE -
testcase_07 TLE -
testcase_08 TLE -
testcase_09 TLE -
testcase_10 TLE -
testcase_11 TLE -
testcase_12 TLE -
testcase_13 TLE -
testcase_14 TLE -
testcase_15 TLE -
testcase_16 TLE -
testcase_17 TLE -
testcase_18 TLE -
testcase_19 TLE -
testcase_20 TLE -
testcase_21 TLE -
testcase_22 TLE -
testcase_23 TLE -
testcase_24 TLE -
testcase_25 TLE -
testcase_26 TLE -
testcase_27 AC 39 ms
53,020 KB
testcase_28 AC 37 ms
53,544 KB
testcase_29 TLE -
testcase_30 TLE -
testcase_31 AC 1,990 ms
259,120 KB
testcase_32 AC 1,999 ms
258,284 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

def _fourier(f, inverse = False):
    f = f[:]
    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 convolution(f, g):
    return _fourier([a * b for a, b in zip(_fourier(f), _fourier(g))], inverse = 1)

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
    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)
resolve()
0