結果

問題 No.1142 XOR と XOR
ユーザー 草苺奶昔草苺奶昔
提出日時 2023-03-11 10:52:31
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 178 ms / 2,000 ms
コード長 1,267 bytes
コンパイル時間 1,543 ms
コンパイル使用メモリ 81,380 KB
実行使用メモリ 136,588 KB
最終ジャッジ日時 2023-10-18 09:46:29
合計ジャッジ時間 5,449 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 100 ms
77,772 KB
testcase_01 AC 95 ms
77,800 KB
testcase_02 AC 96 ms
77,800 KB
testcase_03 AC 177 ms
136,588 KB
testcase_04 AC 157 ms
118,688 KB
testcase_05 AC 147 ms
110,288 KB
testcase_06 AC 160 ms
119,252 KB
testcase_07 AC 169 ms
130,784 KB
testcase_08 AC 178 ms
134,132 KB
testcase_09 AC 176 ms
134,120 KB
testcase_10 AC 172 ms
134,128 KB
testcase_11 AC 99 ms
77,888 KB
testcase_12 AC 96 ms
77,796 KB
testcase_13 AC 96 ms
77,800 KB
testcase_14 AC 136 ms
105,432 KB
testcase_15 AC 131 ms
98,576 KB
testcase_16 AC 99 ms
78,104 KB
testcase_17 AC 141 ms
103,736 KB
testcase_18 AC 104 ms
83,280 KB
testcase_19 AC 163 ms
121,504 KB
testcase_20 AC 137 ms
102,116 KB
testcase_21 AC 122 ms
94,420 KB
testcase_22 AC 110 ms
83,780 KB
testcase_23 AC 159 ms
118,152 KB
testcase_24 AC 167 ms
125,240 KB
testcase_25 AC 136 ms
105,144 KB
testcase_26 AC 157 ms
120,096 KB
testcase_27 AC 141 ms
104,460 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