結果
問題 | No.1066 #いろいろな色 / Red and Blue and more various colors (Easy) |
ユーザー |
![]() |
提出日時 | 2020-10-07 23:09:02 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 716 ms / 2,000 ms |
コード長 | 653 bytes |
コンパイル時間 | 247 ms |
コンパイル使用メモリ | 82,256 KB |
実行使用メモリ | 67,712 KB |
最終ジャッジ日時 | 2024-07-20 04:19:31 |
合計ジャッジ時間 | 6,567 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
import sys input=sys.stdin.readline def I(): return int(input()) def MI(): return map(int, input().split()) def LI(): return list(map(int, input().split())) def main(): mod=998244353 N,Q=MI() A=LI() B=LI() dp=[0]*(N+1) # dp[i][j]はiまで見て,j個だけ色1の玉を持つ dp[0]=1 for i in range(N): for j in range(N-1,-1,-1): dp[j+1]=( dp[j]+ dp[j+1]*(A[i]-1) )%mod dp[0]=(dp[0] * (A[i]-1))%mod # print(dp) for i in range(Q): j=B[i] print(dp[j]) main()