結果

問題 No.2005 Sum of Power Sums
ユーザー chineristACchineristAC
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff # 

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)
0