結果

問題 No.2137 Stairs of Permutation
ユーザー shobonvip
提出日時 2022-11-25 23:17:54
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 551 ms / 2,000 ms
コード長 757 bytes
コンパイル時間 284 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 217,856 KB
最終ジャッジ日時 2024-10-02 05:48:00
合計ジャッジ時間 10,665 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 998244353
N = 10**7 + 5
fact = [1]*(N+1)
factinv = [1]*(N+1)

for i in range(2, N+1):
	fact[i] = fact[i-1] * i % mod

factinv[-1] = pow(fact[-1], mod-2, mod)
for i in range(N-1, 1, -1):
	factinv[i] = factinv[i+1] * (i+1) % mod

def cmb(a, b):
	if (a < b) or (b < 0):
		return 0
	return fact[a] * factinv[b] % mod * factinv[a-b] % mod

n = int(input())

h20, h21 = 0, 2
h30, h31, h32 = 0, 0, 6

# h30^3 + 3h20^2 + h10 の累積和
ans = 0

for i in range(1, n+1):
	ans += (h30 + 3 * h20 + fact[i-1]) % mod * factinv[i] % mod
	ans %= mod
	h20, h21 = h21, (h21 * (2*i+1) % mod - h20 * i % mod * i) % mod
	h30, h31, h32 = h31, h32, (h32 * (3+3*i) % mod - h31 * (3*i*i%mod+3*i+1) % mod + h30 * i % mod * i % mod * i % mod) % mod

print(ans * fact[n] % mod)
0