結果
| 問題 |
No.1282 Display Elements
|
| コンテスト | |
| ユーザー |
norioc
|
| 提出日時 | 2025-06-11 09:59:29 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 279 ms / 2,000 ms |
| コード長 | 1,520 bytes |
| コンパイル時間 | 706 ms |
| コンパイル使用メモリ | 82,176 KB |
| 実行使用メモリ | 169,560 KB |
| 最終ジャッジ日時 | 2025-06-11 09:59:36 |
| 合計ジャッジ時間 | 5,271 ms |
|
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| other | AC * 24 |
ソースコード
class Compression:
def __init__(self, iterable):
self.vs = sorted(set(iterable))
self.v2i = {}
for i, val in enumerate(self.vs):
self.v2i[val] = i
def __len__(self):
return len(self.vs)
def index(self, val):
"""val のインデックスを返す"""
return self.v2i[val]
def value(self, index):
"""インデックスに対応する値を返す"""
return self.vs[index]
def map(self, iterable):
return [self.index(x) for x in iterable]
class FenwickTree:
def __init__(self, n: int):
self.data = [0] * (n+10)
self.n = (n+10)
def add(self, p: int, x: int):
assert 0 <= p < self.n
p += 1
while p < len(self.data):
self.data[p] += x
p += p & -p
def sum(self, p: int) -> int:
"""区間 [0, p] の和"""
assert 0 <= p < self.n
p += 1
s = 0
while p > 0:
s += self.data[p]
p -= p & -p
return s
def rangesum(self, l: int, r: int) -> int:
"""区間 [l, r] の和"""
assert 0 <= l <= r < self.n
s = self.sum(r)
if l > 0:
s -= self.sum(l-1)
return s
N = int(input())
A = list(map(int, input().split()))
B = list(map(int, input().split()))
comp = Compression([-1] + A + B)
ft = FenwickTree(len(comp)+10)
ans = 0
for a, b in zip(comp.map(sorted(A)), comp.map(B)):
ft.add(b, 1)
ans += ft.sum(a-1)
print(ans)
norioc