結果
問題 |
No.1604 Swap Sort:ONE
|
ユーザー |
|
提出日時 | 2022-05-23 23:07:59 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 54 ms / 2,000 ms |
コード長 | 782 bytes |
コンパイル時間 | 184 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 11,136 KB |
最終ジャッジ日時 | 2024-09-20 13:23:21 |
合計ジャッジ時間 | 2,426 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
class Fenwick_Tree: def __init__(self, n): self.n = n self.data = [0] * n def add(self, p, x): p += 1 while p <= self.n: self.data[p - 1] += x p += p & -p def sum(self, l, r): '''範囲[l, r)(lからr-1まで)の総和を求める''' return self._sum(r) - self._sum(l) def _sum(self, r): '''範囲[0, r)(0からr-1まで)の総和を求める''' s = 0 while r > 0: s += self.data[r - 1] r -= r & -r return s n = int(input()) A = list(map(int, input().split())) B = [-1 for _ in range(n)] for i in range(n): B[A[i] - 1] = i FT = Fenwick_Tree(n) ans = 0 for i in range(n): ans += FT.sum(B[i], n) FT.add(B[i], 1) print(ans)