結果
問題 | No.1300 Sum of Inversions |
ユーザー |
|
提出日時 | 2021-02-02 02:32:05 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 861 ms / 2,000 ms |
コード長 | 2,062 bytes |
コンパイル時間 | 146 ms |
コンパイル使用メモリ | 82,192 KB |
実行使用メモリ | 212,096 KB |
最終ジャッジ日時 | 2024-06-29 23:33:58 |
合計ジャッジ時間 | 24,555 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
ソースコード
#region Header#!/usr/bin/env python3# from typing import *import sysimport ioimport mathimport collectionsimport decimalimport itertoolsimport bisectimport heapqdef input():return sys.stdin.readline()[:-1]sys.setrecursionlimit(1000000)#endregion# _INPUT = """10# 3 1 4 1 5 9 2 6 5 3# """# sys.stdin = io.StringIO(_INPUT)MOD = 998244353class BIT:"""Binary Indexed Tree (Fenwick Tree), 1-indexed"""def __init__(self, n):"""Parameters----------n : int要素数。index は 0..n になる。"""self.size = nself.data = [0] * (n+1)# self.depth = n.bit_length()def add(self, i, x):while i <= self.size:self.data[i] += xi += i & -idef get_sum(self, i):s = 0while i > 0:s += self.data[i]i -= i & -ireturn sdef get_rsum(self, l, r):"""[l, r) の sum"""return self.get_sum(r) - self.get_sum(l-1)def main():N = int(input())A = list(map(int, input().split()))T = {a: i for i, a in enumerate(sorted(set(A)))}# 真ん中の A_j を固定# A_i > A_j となる A_iN1 = [0] * NL1 = [0] * Nbit11 = BIT(N)bit12 = BIT(N)for i, a in reversed(list(enumerate(A))):N1[i] = bit11.get_sum(T[a]-1+1)L1[i] = bit12.get_sum(T[a]-1+1)bit11.add(T[a]+1, 1)bit12.add(T[a]+1, a)# A_j > A_k となる A_kN2 = [0] * NL2 = [0] * Nbit21 = BIT(N)bit22 = BIT(N)for i, a in enumerate(A):N2[i] = bit21.get_rsum(T[a]+1+1, N)L2[i] = bit22.get_rsum(T[a]+1+1, N)bit21.add(T[a]+1, 1)bit22.add(T[a]+1, a)ans = 0for i in range(N):ans = (ans + N1[i] * L2[i] + N2[i] * L1[i] + (N1[i] * N2[i]) * A[i]) % MODprint(ans)if __name__ == '__main__':main()