結果

問題 No.2327 Inversion Sum
ユーザー ntuda
提出日時 2023-06-23 21:53:25
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,321 bytes
コンパイル時間 247 ms
コンパイル使用メモリ 82,852 KB
実行使用メモリ 85,496 KB
最終ジャッジ日時 2024-07-01 01:29:46
合計ジャッジ時間 5,632 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353

N, M = map(int, input().split())
PK = [list(map(int, input().split())) for _ in range(M)]

# N: 処理する区間の長さ
N0 = 2 ** (N + 1).bit_length()
data = [0] * (2 * N0)

# a_k の値を x に更新
def update(k, x):
    k += N0 - 1
    data[k] = x  # 加算にするなら+=に
    while k >= 0:
        k = (k - 1) // 2
        data[k] = data[2 * k + 1] + data[2 * k + 2]

# 区間[l, r)の区間和
def query(l, r):
    s = 0
    L = l + N0;
    R = r + N0
    while L < R:
        if R & 1:
            R -= 1
            s += data[R - 1]
        if L & 1:
            s += data[L - 1]
            L += 1
        L >>= 1;
        R >>= 1
    return s

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 //= 2
ans %= MOD
#print(ans)

PK.sort(key=lambda x: x[1])

cnt = 0
for i, (p, k) in enumerate(PK, start=1):
    update(p, 1)
    cnt += query(p + 1, N + 1)
    cnt %= MOD

cnt *= fact[N - M]
ans += cnt
ans %= MOD
#print(ans)

tmp3 = 0
for i, (p, k) in enumerate(PK, start=1):
    tmp = k - i
    tmp2 = N - M - tmp
    tmp3 += (N - p - query(p + 1, N + 1)) * tmp
    tmp3 += (p - query(0, p + 1)) * tmp2
    tmp3 %= MOD
tmp3 *= fact[N - M - 1]
tmp3 %= MOD
ans += tmp3
ans %= MOD
print(ans)
0