結果

問題 No.2137 Stairs of Permutation
ユーザー lam6er
提出日時 2025-03-31 17:47:12
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 813 bytes
コンパイル時間 328 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 123,324 KB
最終ジャッジ日時 2025-03-31 17:47:58
合計ジャッジ時間 5,305 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 1 TLE * 1 -- * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353

N = int(input())

if N == 0:
    print(0)
    exit()

# Precompute inverses
inv = [0] * (N + 1)
for i in range(1, N + 1):
    inv[i] = pow(i, MOD - 2, MOD)

S1 = 0
S2 = 0
S3 = 0
for i in range(1, N + 1):
    S1 = (S1 + inv[i]) % MOD
    inv_i_sq = (inv[i] * inv[i]) % MOD
    S2 = (S2 + inv_i_sq) % MOD
    inv_i_cu = (inv_i_sq * inv[i]) % MOD
    S3 = (S3 + inv_i_cu) % MOD

# Compute term = S1^3 + 3*S1^2 + S1 - 3*S2*(1 + S1) + 2*S3 mod MOD
term = pow(S1, 3, MOD)
term = (term + 3 * pow(S1, 2, MOD)) % MOD
term = (term + S1) % MOD

temp = (3 * S2) % MOD
temp = (temp * (1 + S1)) % MOD
term = (term - temp) % MOD  # Ensure non-negative
term = (term + 2 * S3) % MOD

# Compute factorial mod MOD
fact = 1
for i in range(1, N + 1):
    fact = (fact * i) % MOD

ans = (fact * term) % MOD

print(ans)
0