結果
問題 |
No.1240 Or Sum of Xor Pair
|
ユーザー |
![]() |
提出日時 | 2020-10-24 20:43:09 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
(最新)
AC
(最初)
|
実行時間 | - |
コード長 | 888 bytes |
コンパイル時間 | 317 ms |
コンパイル使用メモリ | 81,996 KB |
実行使用メモリ | 260,212 KB |
最終ジャッジ日時 | 2024-07-21 15:42:20 |
合計ジャッジ時間 | 48,329 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 TLE * 2 |
ソースコード
import sys input = sys.stdin.buffer.readline 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) 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)