結果
問題 | No.2327 Inversion Sum |
ユーザー |
![]() |
提出日時 | 2023-06-27 22:49:37 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 211 ms / 2,000 ms |
コード長 | 1,077 bytes |
コンパイル時間 | 300 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 83,712 KB |
最終ジャッジ日時 | 2024-07-04 15:17:27 |
合計ジャッジ時間 | 4,820 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
MOD = 998244353 class Bit: def __init__(self, n): self.size = n self.tree = [0] * (n + 1) def sum(self, i): s = 0 while i > 0: s += self.tree[i] i -= i & -i return s def add(self, i, x): while i <= self.size: self.tree[i] += x i += i & -i N, M = map(int, input().split()) PK = [list(map(int, input().split())) for _ in range(M)] fact = [1] * (N + 1) r = 1 for i in range(1, N + 1): fact[i] = r = r * i % MOD ans = (N - M) * (N - M - 1) // 2 ans %= MOD ans *= fact[N - M] ans *= pow(2, MOD - 2, MOD) ans %= MOD PK.sort(key=lambda x: x[1]) bit = Bit(N) cnt = 0 for i, (p, k) in enumerate(PK, start=1): bit.add(p, 1) cnt += i - bit.sum(p) cnt %= MOD cnt *= fact[N - M] ans += cnt ans %= MOD tmp3 = 0 for i, (p, k) in enumerate(PK, start=1): tmp = k - i tmp2 = N - M - tmp tmp3 += (N - p - M + bit.sum(p)) * tmp tmp3 += (p - bit.sum(p)) * tmp2 tmp3 %= MOD tmp3 *= fact[N - M - 1] tmp3 %= MOD ans += tmp3 ans %= MOD print(ans)