結果

問題 No.1240 Or Sum of Xor Pair
コンテスト
ユーザー vwxyz
提出日時 2023-12-16 05:17:48
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,575 bytes
コンパイル時間 557 ms
コンパイル使用メモリ 82,388 KB
実行使用メモリ 149,548 KB
最終ジャッジ日時 2024-09-27 07:23:53
合計ジャッジ時間 23,413 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 3
other WA * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
readline=sys.stdin.readline

def Hadamard(polynomial,n,mod=0,inverse=False):
    assert mod%2
    polynomial_=[x for x in polynomial]+[0]*((1<<n)-len(polynomial))
    for bit in range(n):
        for i in range(1<<n):
            ii=i^(1<<bit)
            if i>ii:
                continue
            polynomial_[i],polynomial_[ii]=polynomial_[i]+polynomial_[ii],polynomial_[i]-polynomial_[ii]
            if mod:
                polynomial_[i]%=mod
                polynomial_[ii]%=mod
    if inverse:
        if mod:
            inve_2=pow((mod+1)//2,n)
            for i in range(1<<n):
                polynomial_[i]*=inve_2
                polynomial_[i]%=mod
        else:
            pow_2=pow(2,n)
            for i in range(1<<n):
                polynomial_[i]/=pow_2
    return polynomial_

def XOR_Convolution(polynomial0,polynomial1,mod=0):
    n=(max(len(polynomial0),len(polynomial1))-1).bit_length()
    Hadamard_polynomial0=Hadamard(polynomial0,n,mod=mod)
    Hadamard_polynomial1=Hadamard(polynomial1,n,mod=mod)
    if mod:
        convolution=[x*y%mod for x,y in zip(Hadamard_polynomial0,Hadamard_polynomial1)]
    else:
        convolution=[x*y for x,y in zip(Hadamard_polynomial0,Hadamard_polynomial1)]
    convolution=Hadamard(convolution,n,mod=mod,inverse=True)
    return convolution

N,X=map(int,readline().split())
A=list(map(int,readline().split()))
M=18
mod=(1<<61)-1
ans=0
C=[[0]*(1<<M) for m in range(M+1)]
for a in A:
    C[M][a]+=1
    for m in range(M):
        if a&1<<m:
            C[m][a]+=1
XOR_Convolution(C[M],C[M],mod=mod)
0