結果

問題 No.1939 Numbered Colorful Balls
ユーザー tassei903tassei903
提出日時 2021-12-12 00:24:48
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 232 ms / 2,000 ms
コード長 728 bytes
コンパイル時間 607 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 65,024 KB
最終ジャッジ日時 2024-07-06 02:39:45
合計ジャッジ時間 4,800 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 225 ms
64,640 KB
testcase_01 AC 45 ms
59,904 KB
testcase_02 AC 134 ms
61,440 KB
testcase_03 AC 232 ms
64,768 KB
testcase_04 AC 34 ms
52,224 KB
testcase_05 AC 229 ms
64,384 KB
testcase_06 AC 132 ms
65,024 KB
testcase_07 AC 75 ms
62,208 KB
testcase_08 AC 56 ms
61,568 KB
testcase_09 AC 143 ms
63,616 KB
testcase_10 AC 46 ms
61,184 KB
testcase_11 AC 91 ms
62,720 KB
testcase_12 AC 54 ms
61,696 KB
testcase_13 AC 48 ms
61,184 KB
testcase_14 AC 79 ms
62,208 KB
testcase_15 AC 63 ms
61,952 KB
testcase_16 AC 138 ms
63,488 KB
testcase_17 AC 107 ms
62,464 KB
testcase_18 AC 81 ms
62,336 KB
testcase_19 AC 139 ms
63,360 KB
testcase_20 AC 68 ms
62,208 KB
testcase_21 AC 119 ms
63,616 KB
testcase_22 AC 51 ms
61,824 KB
testcase_23 AC 91 ms
64,896 KB
testcase_24 AC 119 ms
63,360 KB
testcase_25 AC 84 ms
62,720 KB
testcase_26 AC 213 ms
64,640 KB
testcase_27 AC 35 ms
52,224 KB
testcase_28 AC 33 ms
52,352 KB
testcase_29 AC 97 ms
64,768 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = lambda :sys.stdin.readline()[:-1]
ni = lambda :int(input())
na = lambda :list(map(int,input().split()))
sys.setrecursionlimit(10**7)
yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES")
no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO")
#######################################################################

n,m = na()
l = na()
dp = [0]*(n+1)

dp[0] = 1
mod = 998244353

for k in range(1,n+1):
    for j in l:
        if k-j>=0:
            dp[k]+=dp[k-j]*j%mod*(n+1)%mod
            dp[k]%=mod
        if k-j>=0:
            dp[k]-=dp[k-j]*(k-j)%mod
            dp[k]%=mod
    dp[k]*=pow(k,mod-2,mod)
    dp[k]%=mod
print(dp[n]*pow(n+1,mod-2,mod)%mod)
0