結果

問題 No.1066 #いろいろな色 / Red and Blue and more various colors (Easy)
コンテスト
ユーザー umezo
提出日時 2020-05-29 23:29:12
言語 C++17
(gcc 13.3.0 + boost 1.89.0)
結果
MLE  
実行時間 -
コード長 550 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,270 ms
コンパイル使用メモリ 192,500 KB
最終ジャッジ日時 2025-01-10 18:28:18
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample MLE * 3
other MLE * 24
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#define rep(i, n) for (int i = 0; i < (int)(n); i++)
typedef long long ll;

#include <bits/stdc++.h>
using namespace std;

#define MOD 998244353

int main(){
  ll n,q;
  cin>>n>>q;
  
  ll A[6001];
  A[0]=0;
  rep(i,n) cin>>A[i+1];
  
  ll dp[6001][6001];
  rep(i,n+1) dp[i][i]=1;
  for(int i=1;i<=n;i++) dp[i][0]=dp[i-1][0]*(A[i]-1)%MOD;     
     
  for(int i=2;i<=n;i++){
    for(int j=1;j<i;j++){
      dp[i][j]=(dp[i-1][j-1]+dp[i-1][j]*(A[i]-1)%MOD)%MOD;
    }
  }
  
  int x;
  rep(i,q){
    cin>>x;
    cout<<dp[n][x]<<endl;
  }

  return 0;
}
0