結果
問題 | No.1066 #いろいろな色 / Red and Blue and more various colors (Easy) |
ユーザー |
![]() |
提出日時 | 2020-05-29 22:29:37 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 68 ms / 2,000 ms |
コード長 | 1,230 bytes |
コンパイル時間 | 1,691 ms |
コンパイル使用メモリ | 172,292 KB |
実行使用メモリ | 6,820 KB |
最終ジャッジ日時 | 2024-11-06 05:51:56 |
合計ジャッジ時間 | 3,227 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
#include <bits/stdc++.h> using namespace std; using ll=long long; using vin=vector<int>; using vll=vector<long long>; using vvin=vector<vector<int>>; using vvll=vector<vector<long long>>; using vstr=vector<string>; using vvstr=vector<vector<string>>; using vch=vector<char>; using vvch=vector<vector<char>>; using vbo=vector<bool>; using vvbo=vector<vector<bool>>; using vpii=vector<pair<int,int>>; using pqsin=priority_queue<int,vector<int>,greater<int>>; #define mp make_pair #define rep(i,n) for(int i=0;i<(int)(n);i++) #define rep2(i,s,n) for(int i=(s);i<(int)(n);i++) #define all(v) v.begin(),v.end() #define decp(n) cout<<fixed<<setprecision((int)n) const int inf=1e9+7; const ll INF=1e18; const int mod=998244353; int main(){ int n,q;cin>>n>>q; vll a(n);rep(i,n)cin>>a[i]; vvll dp(2,vll(n+1));//dp[i][j]:i個目の箱まで殻球を一個ずつ取り出して色1がj個含まれる場合の数 dp[0][0]=1; rep2(i,1,n+1)rep(j,n+1){ if(j>i)continue; dp[i%2][j]=(a[i-1]-1)*dp[(i-1)%2][j]; if(j>0)dp[i%2][j]+=dp[(i-1)%2][j-1]; dp[i%2][j]%=mod; } int b; vll ans(q); rep(i,q){ cin>>b; ans[i]=dp[n%2][b]; } rep(i,q)cout<<ans[i]<<endl; }