結果

問題 No.1066 #いろいろな色 / Red and Blue and more various colors (Easy)
ユーザー 👑 Kazun
提出日時 2020-06-10 15:24:42
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 398 bytes
コンパイル時間 848 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 359,108 KB
最終ジャッジ日時 2024-06-23 06:12:02
合計ジャッジ時間 6,547 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 13 MLE * 11
権限があれば一括ダウンロードができます

ソースコード

diff # 

#--------------------------------------
N,Q=map(int,input().split())
A=list(map(int,input().split()))
B=list(map(int,input().split()))
M=998244353

D=[[0]*(N+1) for _ in range(N+1)] #D[x][k]:xまで,k個の色1

D[0][0]=1

for x in range(N):
    D[x+1][0]=((A[x]-1)*D[x][0])%M
    for k in range(N):
        D[x+1][k+1]=(D[x][k]+(A[x]-1)*D[x][k+1])%M

for b in B:
    print(D[-1][b])
0