結果
問題 | No.2369 Some Products |
ユーザー | momoyuu |
提出日時 | 2023-10-26 12:22:11 |
言語 | C++23 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 143 ms / 2,500 ms |
コード長 | 1,916 bytes |
コンパイル時間 | 2,876 ms |
コンパイル使用メモリ | 255,168 KB |
実行使用メモリ | 69,120 KB |
最終ジャッジ日時 | 2024-09-25 12:22:31 |
合計ジャッジ時間 | 4,772 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,376 KB |
testcase_02 | AC | 142 ms
69,120 KB |
testcase_03 | AC | 143 ms
68,992 KB |
testcase_04 | AC | 142 ms
68,992 KB |
testcase_05 | AC | 15 ms
7,680 KB |
testcase_06 | AC | 126 ms
68,992 KB |
testcase_07 | AC | 12 ms
8,192 KB |
testcase_08 | AC | 28 ms
15,616 KB |
testcase_09 | AC | 13 ms
6,784 KB |
testcase_10 | AC | 55 ms
26,496 KB |
testcase_11 | AC | 94 ms
53,120 KB |
testcase_12 | AC | 8 ms
5,504 KB |
testcase_13 | AC | 64 ms
32,000 KB |
testcase_14 | AC | 70 ms
37,376 KB |
ソースコード
#include<bits/stdc++.h> using namespace std; using ll = long long; #include<atcoder/modint> using mint = atcoder::modint998244353; mint ans[5010]; mint p[5010]; int le[5010],ri[5010],k[5010]; void solve(int l,int r,vector<int>&idx){ if(r-l==1){ for(int i = 0;i<idx.size();i++){ int ni = idx[i]; ans[ni] = p[l]; } return; } int mid = (r+l)>>1; vector<vector<mint>> left(mid-l+1,vector<mint>(mid-l+1,0)); vector<vector<mint>> right(r-mid+1,vector<mint>(r-mid+1,0)); left[mid-l][0] = 1; for(int i = mid-1;i>=l;i--){ int ni = i - l; for(int j = mid-l;j>=0;j--){ left[ni][j] = left[ni+1][j]; if(j-1>=0) left[ni][j] += left[ni+1][j-1] * p[i]; } } right[0][0] = 1; for(int i = mid;i<r;i++){ int ni = i - mid + 1; for(int j = r-mid;j>=0;j--){ right[ni][j] = right[ni-1][j]; if(j-1>=0) right[ni][j] += right[ni-1][j-1] * p[i]; } } vector<int> ls,rs; for(int i = 0;i<idx.size();i++){ int ni = idx[i]; if(ri[ni]<=mid) ls.push_back(ni); else if(mid<=le[ni]) rs.push_back(ni); else{ for(int j = 0;j<=k[ni];j++){ int nni = le[ni] - l; int nnj = ri[ni] - mid; if(j>=mid-l+1||k[ni]-j>=r-mid+1) continue; ans[ni] += left[nni][j] * right[nnj][k[ni]-j]; } } } solve(l,mid,ls); solve(mid,r,rs); } int main(){ cin.tie(nullptr); ios::sync_with_stdio(false); int n; cin>>n; for(int i = 0;i<n;i++){ ll a; cin>>a; p[i] = a; } int q; cin>>q; vector<int> idx(q); for(int i = 0;i<q;i++){ idx[i] = i; cin>>le[i]>>ri[i]>>k[i]; le[i]--; } solve(0,n,idx); for(int i = 0;i<q;i++) cout<<ans[i].val()<<endl; }