結果

問題 No.1863 Xor Sum 2...?
ユーザー tamato
提出日時 2022-03-04 22:45:59
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 116 ms / 2,000 ms
コード長 1,192 bytes
コンパイル時間 233 ms
コンパイル使用メモリ 82,588 KB
実行使用メモリ 103,820 KB
最終ジャッジ日時 2024-07-18 21:18:56
合計ジャッジ時間 3,789 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff #

mod = 1000000007
eps = 10**-9


def main():
    import sys
    input = sys.stdin.readline

    N = int(input())
    A = list(map(int, input().split()))
    B = list(map(int, input().split()))

    cs_A = [0] * (N + 1)
    cxor_A = [0] * (N + 1)
    for i, a in enumerate(A):
        cs_A[i + 1] = cs_A[i] + a
        cxor_A[i + 1] = cxor_A[i] ^ a

    cxor_B = [0] * (N + 1)
    for i, b in enumerate(B):
        cxor_B[i + 1] = cxor_B[i] ^ b
    cs_B = [0] * (N + 1)
    for i in range(N):
        cs_B[i + 1] = cs_B[i] + cxor_B[i + 1]

    ans = 0
    for i in range(N):
        ok = i
        ng = N
        mid = (ok + ng) // 2
        while ng - ok > 1:
            c = cs_A[mid + 1] - cs_A[i]
            x = cxor_A[mid + 1] ^ cxor_A[i]
            if c == x:
                ok = mid
            else:
                ng = mid
            mid = (ok + ng) // 2

        if B[i] == 0:
            ans += 1
        if ok != i:
            if cxor_B[i] == 1:
                ans += cs_B[ok + 1] - cs_B[i + 1]
            else:
                ans += (ok - i) - (cs_B[ok + 1] - cs_B[i + 1])
        #print(i, ok, ans)
    print(ans)
    #print(cs_B)


if __name__ == '__main__':
    main()
0