結果

問題 No.2936 Sum of Square of Mex
コンテスト
ユーザー hirayuu_yc
提出日時 2024-10-12 10:01:46
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 286 bytes
コンパイル時間 935 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 264,320 KB
最終ジャッジ日時 2024-10-12 10:01:51
合計ジャッジ時間 4,187 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 TLE * 1
other -- * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD=998244353
N,M=map(int,input().split())
ans=0
for i in range(min(N,M+1)+1):
	dp=[1]
	flg=0
	if i==M+1:
		flg=1
	for j in range(N):
		ndp=[0]*(j+2)
		for k in range(j+1):
			dp[k]%=MOD
			ndp[k]+=dp[k]*(M-i+k+flg)
			ndp[k+1]+=dp[k]*(i-k)
		dp=ndp
	ans+=dp[i]*i*i
	ans%=MOD
print(ans)
0