結果
問題 | No.1300 Sum of Inversions |
ユーザー | mkawa2 |
提出日時 | 2020-11-27 22:15:26 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 421 ms / 2,000 ms |
コード長 | 1,901 bytes |
コンパイル時間 | 345 ms |
コンパイル使用メモリ | 82,436 KB |
実行使用メモリ | 172,452 KB |
最終ジャッジ日時 | 2024-07-26 13:12:49 |
合計ジャッジ時間 | 13,001 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
ソースコード
import syssys.setrecursionlimit(10**6)int1 = lambda x: int(x)-1p2D = lambda x: print(*x, sep="\n")def II(): return int(sys.stdin.buffer.readline())def MI(): return map(int, sys.stdin.buffer.readline().split())def LI(): return list(map(int, sys.stdin.buffer.readline().split()))def LLI(rows_number): return [LI() for _ in range(rows_number)]def MI1(): return map(int1, sys.stdin.buffer.readline().split())def LI1(): return list(map(int1, sys.stdin.buffer.readline().split()))def LLI1(rows_number): return [LI1() for _ in range(rows_number)]def BI(): return sys.stdin.buffer.readline().rstrip()def SI(): return sys.stdin.buffer.readline().rstrip().decode()def inval(ni, nj, h, w):if ni < 0 or ni >= h or nj < 0 or nj >= w: return Truereturn Falsedij = [(0, 1), (-1, 0), (0, -1), (1, 0)]inf = 10**16# md = 10 ** 9 + 7md = 998244353class BitSum:def __init__(self, n):self.n = n+1self.table = [0]*self.ndef add(self, i, x):i += 1while i < self.n:self.table[i] += xi += i & -i# [0,i]の和def sum(self, i):i += 1res = 0while i > 0:res += self.table[i]i -= i & -ireturn resn=II()aa=LI()enc= {a:i for i,a in enumerate(sorted(set(aa)))}bb=[enc[a] for a in aa]ans=0cc=[0]*nlow=[0]*nup=[0]*nbit1=BitSum(n+3)bit2=BitSum(n+3)for i,b in enumerate(bb):b=n-bc=bit2.sum(b-1)cc[i]+=cans%=mdc=bit1.sum(b-1)up[i]=cbit2.add(b,c)bit1.add(b,1)# print(cc)bit1=BitSum(n)bit2=BitSum(n)for i,b in enumerate(bb[::-1]):c=bit2.sum(b-1)cc[n-1-i]+=cans%=mdc=bit1.sum(b-1)low[n-1-i]=cbit2.add(b,c)bit1.add(b,1)# print(cc)# print(up)# print(low)for i,(u,l) in enumerate(zip(up,low)):cc[i]+=u*lfor a,c in zip(aa,cc):ans+=a*cans%=mdprint(ans)