結果

問題 No.1142 XOR と XOR
ユーザー 草苺奶昔草苺奶昔
提出日時 2023-03-11 10:52:31
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 169 ms / 2,000 ms
コード長 1,267 bytes
コンパイル時間 298 ms
コンパイル使用メモリ 81,848 KB
実行使用メモリ 136,872 KB
最終ジャッジ日時 2024-09-18 06:16:08
合計ジャッジ時間 4,913 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 89 ms
78,464 KB
testcase_01 AC 86 ms
78,336 KB
testcase_02 AC 94 ms
78,308 KB
testcase_03 AC 169 ms
136,872 KB
testcase_04 AC 144 ms
119,148 KB
testcase_05 AC 141 ms
110,452 KB
testcase_06 AC 151 ms
119,428 KB
testcase_07 AC 151 ms
131,044 KB
testcase_08 AC 153 ms
134,836 KB
testcase_09 AC 160 ms
134,668 KB
testcase_10 AC 161 ms
134,760 KB
testcase_11 AC 91 ms
78,256 KB
testcase_12 AC 85 ms
77,824 KB
testcase_13 AC 85 ms
78,080 KB
testcase_14 AC 122 ms
105,832 KB
testcase_15 AC 119 ms
98,804 KB
testcase_16 AC 88 ms
78,080 KB
testcase_17 AC 127 ms
104,560 KB
testcase_18 AC 92 ms
83,532 KB
testcase_19 AC 144 ms
122,108 KB
testcase_20 AC 125 ms
102,068 KB
testcase_21 AC 108 ms
95,228 KB
testcase_22 AC 95 ms
84,476 KB
testcase_23 AC 136 ms
118,400 KB
testcase_24 AC 153 ms
126,548 KB
testcase_25 AC 121 ms
105,648 KB
testcase_26 AC 140 ms
120,728 KB
testcase_27 AC 125 ms
104,644 KB
権限があれば一括ダウンロードができます

ソースコード

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] + MOD - 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