結果

問題 No.1240 Or Sum of Xor Pair
ユーザー chineristAC
提出日時 2020-09-26 00:32:06
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,601 bytes
コンパイル時間 164 ms
コンパイル使用メモリ 82,028 KB
実行使用メモリ 278,604 KB
最終ジャッジ日時 2024-06-28 08:29:50
合計ジャッジ時間 6,797 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample TLE * 1 -- * 2
other -- * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

def xorconv(n,X,Y,mod=0):
    if mod:
        if n==0:
            res=[(X[0]*Y[0])%mod]
            return res
        bit = 1<<(n-1)
        x=[X[i]+X[i^bit] for i in range(2**(n-1))]
        y=[Y[i]+Y[i^bit] for i in range(2**(n-1))]
        z=[X[i]-X[i^bit] for i in range(2**(n-1))]
        w=[Y[i]-Y[i^bit] for i in range(2**(n-1))]
        res1=xorconv(n-1,x,y)
        res2=xorconv(n-1,z,w)
        former=[(res1[i]+res2[i])*inv for i in range(2**(n-1))]
        latter=[(res1[i]-res2[i])*inv for i in range(2**(n-1))]
        former=list(map(lambda x:x%mod,former))
        latter=list(map(lambda x:x%mod,latter))
        return former+latter
    else:
        if n==0:
            res=[(X[0]*Y[0])]
            return res
        bit = 1<<(n-1)
        x=[X[i]+X[i^bit] for i in range(bit)]
        y=[Y[i]+Y[i^bit] for i in range(bit)]
        z=[X[i]-X[i^bit] for i in range(bit)]
        w=[Y[i]-Y[i^bit] for i in range(bit)]
        res1=xorconv(n-1,x,y)
        res2=xorconv(n-1,z,w)
        former=[(res1[i]+res2[i])//2 for i in range(bit)]
        latter=[(res1[i]-res2[i])//2 for i in range(bit)]
        return former+latter


N,X = map(int,input().split())
A = list(map(int,input().split()))

cnt = [0 for i in range(2**18)]
for a in A:
    cnt[a] += 1
cnt = xorconv(18,cnt,cnt)
cnt = sum(cnt[:X])

res = 0
for i in range(18):
    data = [0 for j in range(2**18)]
    for j in range(N):
        if not A[j]>>i & 1:
            data[A[j]] += 1

    data = xorconv(18,data,data)
    plus = 1<<i
    num = cnt - sum(data[:X]) 
    res += plus * num

res -= sum(A)
res //= 2
print(res)
0