結果

問題 No.1142 XOR と XOR
ユーザー 草苺奶昔
提出日時 2023-03-11 10:53:38
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 154 ms / 2,000 ms
コード長 1,260 bytes
コンパイル時間 441 ms
コンパイル使用メモリ 81,600 KB
実行使用メモリ 137,080 KB
最終ジャッジ日時 2024-09-18 06:16:14
合計ジャッジ時間 4,940 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff #

from typing import List


MOD = int(1e9) + 7
INV = pow(2, MOD - 2, MOD)


def xorConvolution(a: List[int], b: List[int]) -> List[int]:
    a, b = a[:], b[:]
    a = _walshHadamardTransform(a, 1)
    b = _walshHadamardTransform(b, 1)
    for i in range(len(a)):
        a[i] = a[i] * b[i]
    res = _walshHadamardTransform(a, INV)
    return res


def _walshHadamardTransform(f, op):
    n = len(f)
    l_, k = 2, 1
    while l_ <= n:
        for i in range(0, n, l_):
            for j in range(k):
                f[i + j], f[i + j + k] = (f[i + j] + f[i + j + k]) * op, (
                    f[i + j] - f[i + j + k]
                ) * op
        l_, k = l_ << 1, k << 1
    return f


if __name__ == "__main__":
    n, m, k = map(int, input().split())
    nums1 = list(map(int, input().split()))
    nums2 = list(map(int, input().split()))

    MAX = 1024
    f, g = [0] * MAX, [0] * MAX
    f[0], g[0] = 1, 1  # 统计前缀xor的频率
    xor_ = 0
    for num in nums1:
        xor_ ^= num
        f[xor_] += 1
    xor_ = 0
    for num in nums2:
        xor_ ^= num
        g[xor_] += 1

    f, g = xorConvolution(f, f), xorConvolution(g, g)
    f[0], g[0] = (f[0] - n - 1), (g[0] - m - 1)

    res = xorConvolution(f, g)
    print((res[k] // 4) % MOD)
0