結果
問題 | No.2005 Sum of Power Sums |
ユーザー |
|
提出日時 | 2022-02-12 22:30:11 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 918 ms / 2,000 ms |
コード長 | 856 bytes |
コンパイル時間 | 219 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 105,600 KB |
最終ジャッジ日時 | 2024-06-12 05:56:31 |
合計ジャッジ時間 | 6,755 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 18 |
ソースコード
mod = 998244353 N,M = map(int,input().split()) k = list(map(int,input().split())) K = max(k) cnt = [0 for i in range(K+1)] for i in range(N): cnt[k[i]] += 1 COMB = [0 for j in range(K+2)] comb = 1 for i in range(N-1): comb *= (M+N-1-i) * pow(i+1,mod-2,mod) % mod comb %= mod COMB[0] = comb for j in range(1,K+2): COMB[j] = COMB[j-1] * (M+N-1+j) * pow(N-1+j,mod-2,mod) % mod COMB[j] %= mod tmp = [0] * (K+3) tmp[-1] = 1 res = 0 for i in range(1,K+1): for j in range(-K-1,0): if not tmp[j]: continue tmp[j-1] = -j*tmp[j] % mod tmp[j] = 0 for j in range(-K-1,0)[::-1]: if not tmp[j]: continue tmp[j+1] -= tmp[j] tmp[j+1] %= mod for j in range(-K-1,0): res += tmp[j] * (COMB[-j] * cnt[i] % mod) % mod res %= mod print(res)