結果
問題 | No.1115 二つの数列 / Two Sequences |
ユーザー |
![]() |
提出日時 | 2020-07-17 22:16:04 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 141 ms / 2,000 ms |
コード長 | 1,260 bytes |
コンパイル時間 | 247 ms |
コンパイル使用メモリ | 81,664 KB |
実行使用メモリ | 109,568 KB |
最終ジャッジ日時 | 2024-11-30 00:42:43 |
合計ジャッジ時間 | 4,743 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 35 |
ソースコード
class BIT:"""一点加算、区間取得クエリをそれぞれO(logN)で答えるデータ構造"""def __init__(self, n):self.size = nself.bit = [0] * (n + 1)def build(self, array):"""arrayを初期値とするBinaryIndexTreeを構築する O(N)"""for i in range(self.size):self.bit[i + 1] = array[i]for i in range(1, self.size):if i + (i & -i) > self.size:continueself.bit[i + (i & -i)] += self.bit[i]def _sum(self, i):s = 0while i > 0:s += self.bit[i]i -= i & -ireturn sdef add(self, i, val):"""i番目の要素にvalを加える"""i += 1while i <= self.size:self.bit[i] += vali += i & -idef get_sum(self, l, r):"""区間[l, r)の和を求める"""return self._sum(r) - self._sum(l)n = int(input())a = list(map(int, input().split()))b = list(map(int, input().split()))mapping = {}for i in range(n):mapping[a[i]] = ic = [0] * nfor i in range(n):c[i] = mapping[b[i]]bit = BIT(n)ans = 0for i in range(n)[::-1]:bit.add(c[i], 1)ans += bit.get_sum(0, c[i])print(ans)