結果
問題 | No.2005 Sum of Power Sums |
ユーザー | chineristAC |
提出日時 | 2022-02-12 22:30:11 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 928 ms / 2,000 ms |
コード長 | 856 bytes |
コンパイル時間 | 312 ms |
コンパイル使用メモリ | 87,148 KB |
実行使用メモリ | 107,088 KB |
最終ジャッジ日時 | 2023-09-02 23:55:06 |
合計ジャッジ時間 | 7,450 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge13 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 71 ms
71,236 KB |
testcase_01 | AC | 70 ms
71,368 KB |
testcase_02 | AC | 412 ms
77,540 KB |
testcase_03 | AC | 71 ms
71,264 KB |
testcase_04 | AC | 70 ms
71,216 KB |
testcase_05 | AC | 70 ms
71,356 KB |
testcase_06 | AC | 88 ms
76,460 KB |
testcase_07 | AC | 81 ms
76,276 KB |
testcase_08 | AC | 72 ms
71,120 KB |
testcase_09 | AC | 115 ms
76,832 KB |
testcase_10 | AC | 70 ms
71,284 KB |
testcase_11 | AC | 202 ms
78,032 KB |
testcase_12 | AC | 71 ms
71,444 KB |
testcase_13 | AC | 203 ms
77,868 KB |
testcase_14 | AC | 299 ms
105,764 KB |
testcase_15 | AC | 927 ms
106,880 KB |
testcase_16 | AC | 928 ms
106,704 KB |
testcase_17 | AC | 868 ms
106,836 KB |
testcase_18 | AC | 73 ms
71,288 KB |
testcase_19 | AC | 203 ms
77,920 KB |
testcase_20 | AC | 874 ms
107,088 KB |
ソースコード
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)