結果

問題 No.1209 XOR Into You
ユーザー 👑 rin204
提出日時 2023-12-29 20:13:48
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 231 ms / 2,000 ms
コード長 1,391 bytes
コンパイル時間 393 ms
コンパイル使用メモリ 82,420 KB
実行使用メモリ 123,012 KB
最終ジャッジ日時 2024-09-27 16:19:09
合計ジャッジ時間 9,016 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff #

class BIT:
    def __init__(self, n):
        self.n = n
        self.data = [0] * (n + 1)
        if n == 0:
            self.n0 = 0
        else:
            self.n0 = 1 << (n.bit_length() - 1)

    def sum_(self, i):
        s = 0
        while i > 0:
            s += self.data[i]
            i -= i & -i
        return s

    def sum(self, l, r=-1):
        if r == -1:
            return self.sum_(l)
        else:
            return self.sum_(r) - self.sum_(l)

    def get(self, i):
        return self.sum(i, i + 1)

    def add(self, i, x):
        i += 1
        while i <= self.n:
            self.data[i] += x
            i += i & -i

    def lower_bound(self, x):
        if x <= 0:
            return 0
        i = 0
        k = self.n0
        while k > 0:
            if i + k <= self.n and self.data[i + k] < x:
                x -= self.data[i + k]
                i += k
            k //= 2
        return i + 1


n = int(input())
A = list(map(int, input().split()))
B = list(map(int, input().split()))
if A[0] != B[0]:
    print(-1)
    exit()


def f(A):
    return [A[i] ^ A[i + 1] for i in range(len(A) - 1)]


A = f(A)
B = f(B)

if sorted(A) != sorted(B):
    print(-1)
    exit()

dic = {}
for i, a in enumerate(A):
    dic.setdefault(a, []).append(i)

bit = BIT(n)
ans = 0
for b in B[::-1]:
    i = dic[b].pop()
    ans += bit.sum(i)
    bit.add(i, 1)

print(ans)
0