結果

問題 No.1066 #いろいろな色 / Red and Blue and more various colors (Easy)
ユーザー umezo
提出日時 2020-05-29 23:17:37
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 566 bytes
コンパイル時間 997 ms
コンパイル使用メモリ 72,300 KB
最終ジャッジ日時 2025-01-10 18:19:15
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 2
other AC * 17 WA * 1 MLE * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#include<iostream>
#include<vector>
using namespace std;

#define MOD 998244353

int main(){
  ll n,q;
  cin>>n>>q;
  
  vector<ll> A(n+1);
  A[0]=0;
  rep(i,n) cin>>A[i+1];
  
  ll dp[n+1][n+1];
  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<n;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