結果
問題 | No.1300 Sum of Inversions |
ユーザー | tktk_snsn |
提出日時 | 2020-11-27 22:26:59 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 955 ms / 2,000 ms |
コード長 | 1,969 bytes |
コンパイル時間 | 252 ms |
コンパイル使用メモリ | 87,224 KB |
実行使用メモリ | 141,492 KB |
最終ジャッジ日時 | 2023-10-09 20:28:52 |
合計ジャッジ時間 | 26,028 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge14 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 96 ms
71,904 KB |
testcase_01 | AC | 94 ms
71,836 KB |
testcase_02 | AC | 95 ms
71,784 KB |
testcase_03 | AC | 775 ms
121,376 KB |
testcase_04 | AC | 756 ms
121,604 KB |
testcase_05 | AC | 641 ms
121,064 KB |
testcase_06 | AC | 861 ms
129,800 KB |
testcase_07 | AC | 788 ms
127,012 KB |
testcase_08 | AC | 893 ms
134,328 KB |
testcase_09 | AC | 890 ms
134,148 KB |
testcase_10 | AC | 543 ms
121,948 KB |
testcase_11 | AC | 546 ms
121,632 KB |
testcase_12 | AC | 738 ms
121,408 KB |
testcase_13 | AC | 737 ms
121,408 KB |
testcase_14 | AC | 955 ms
141,492 KB |
testcase_15 | AC | 883 ms
133,300 KB |
testcase_16 | AC | 778 ms
122,140 KB |
testcase_17 | AC | 537 ms
122,508 KB |
testcase_18 | AC | 598 ms
119,400 KB |
testcase_19 | AC | 681 ms
121,204 KB |
testcase_20 | AC | 707 ms
120,992 KB |
testcase_21 | AC | 705 ms
121,012 KB |
testcase_22 | AC | 638 ms
121,076 KB |
testcase_23 | AC | 845 ms
128,584 KB |
testcase_24 | AC | 648 ms
119,092 KB |
testcase_25 | AC | 579 ms
121,524 KB |
testcase_26 | AC | 571 ms
121,492 KB |
testcase_27 | AC | 628 ms
121,224 KB |
testcase_28 | AC | 924 ms
138,128 KB |
testcase_29 | AC | 687 ms
120,972 KB |
testcase_30 | AC | 895 ms
134,756 KB |
testcase_31 | AC | 640 ms
120,880 KB |
testcase_32 | AC | 673 ms
119,280 KB |
testcase_33 | AC | 400 ms
126,064 KB |
testcase_34 | AC | 407 ms
130,704 KB |
testcase_35 | AC | 481 ms
133,824 KB |
testcase_36 | AC | 503 ms
141,472 KB |
ソースコード
from collections import Counter, defaultdict mod = 998244353 class fenwick_tree(object): def __init__(self, n): self.n = n self.log = n.bit_length() self.data = [0] * n def __sum(self, r): s = 0 while r > 0: s += self.data[r - 1] r -= r & -r return s def add(self, p, x): """ a[p] += xを行う""" p += 1 while p <= self.n: self.data[p - 1] += x p += p & -p def sum(self, l, r): """a[l] + a[l+1] + .. + a[r-1]を返す""" return self.__sum(r) - self.__sum(l) def lower_bound(self, x): """a[0] + a[1] + .. a[i] >= x となる最小のiを返す""" if x <= 0: return -1 i = 0 k = 1 << self.log while k: if i + k <= self.n and self.data[i + k - 1] < x: x -= self.data[i + k - 1] i += k k >>= 1 return i N = int(input()) A = list(map(int, input().split())) memo = defaultdict(list) for i, a in enumerate(A): memo[a].append(i) key = sorted(memo.keys()) # 左にある自分より大きいやつを見る L_sum = [0] * N L_cnt = [0] * N bit_sum = fenwick_tree(N) bit_cnt = fenwick_tree(N) for k in key[::-1]: for i in memo[k]: L_sum[i] = bit_sum.sum(0, i) L_cnt[i] = bit_cnt.sum(0, i) for i in memo[k]: bit_sum.add(i, k) bit_cnt.add(i, 1) # 右にある自分より小さいやつを見る R_sum = [0] * N R_cnt = [0] * N bit_sum = fenwick_tree(N) bit_cnt = fenwick_tree(N) for k in key: for i in memo[k]: R_sum[i] = bit_sum.sum(i, N) R_cnt[i] = bit_cnt.sum(i, N) for i in memo[k]: bit_sum.add(i, k) bit_cnt.add(i, 1) ans = 0 for i in range(1, N - 1): ans += L_sum[i] * R_cnt[i] ans %= mod ans += R_sum[i] * L_cnt[i] ans %= mod ans += A[i] * R_cnt[i] * L_cnt[i] ans %= mod print(ans)