結果

問題 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 88 ms
78,148 KB
testcase_01 AC 85 ms
78,388 KB
testcase_02 AC 86 ms
78,200 KB
testcase_03 AC 153 ms
137,080 KB
testcase_04 AC 136 ms
119,516 KB
testcase_05 AC 134 ms
110,864 KB
testcase_06 AC 142 ms
120,208 KB
testcase_07 AC 151 ms
131,132 KB
testcase_08 AC 152 ms
134,848 KB
testcase_09 AC 153 ms
134,740 KB
testcase_10 AC 154 ms
134,620 KB
testcase_11 AC 83 ms
78,052 KB
testcase_12 AC 84 ms
78,480 KB
testcase_13 AC 86 ms
78,056 KB
testcase_14 AC 120 ms
106,552 KB
testcase_15 AC 111 ms
98,660 KB
testcase_16 AC 86 ms
78,900 KB
testcase_17 AC 124 ms
103,896 KB
testcase_18 AC 91 ms
83,688 KB
testcase_19 AC 147 ms
122,368 KB
testcase_20 AC 129 ms
103,196 KB
testcase_21 AC 112 ms
92,776 KB
testcase_22 AC 96 ms
84,376 KB
testcase_23 AC 143 ms
119,384 KB
testcase_24 AC 152 ms
127,036 KB
testcase_25 AC 114 ms
101,592 KB
testcase_26 AC 139 ms
120,596 KB
testcase_27 AC 128 ms
105,384 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] - 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