結果
問題 | No.1068 #いろいろな色 / Red and Blue and more various colors (Hard) |
ユーザー | fuppy_kyopro |
提出日時 | 2020-05-29 23:13:07 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 536 ms / 3,500 ms |
コード長 | 4,988 bytes |
コンパイル時間 | 2,036 ms |
コンパイル使用メモリ | 171,952 KB |
実行使用メモリ | 31,896 KB |
最終ジャッジ日時 | 2024-04-24 01:09:10 |
合計ジャッジ時間 | 12,400 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 1 ms
5,376 KB |
testcase_02 | AC | 1 ms
5,376 KB |
testcase_03 | AC | 16 ms
5,376 KB |
testcase_04 | AC | 11 ms
5,376 KB |
testcase_05 | AC | 13 ms
5,376 KB |
testcase_06 | AC | 9 ms
5,376 KB |
testcase_07 | AC | 9 ms
5,376 KB |
testcase_08 | AC | 12 ms
5,376 KB |
testcase_09 | AC | 12 ms
5,376 KB |
testcase_10 | AC | 6 ms
5,376 KB |
testcase_11 | AC | 9 ms
5,376 KB |
testcase_12 | AC | 6 ms
5,376 KB |
testcase_13 | AC | 521 ms
31,892 KB |
testcase_14 | AC | 515 ms
31,764 KB |
testcase_15 | AC | 517 ms
31,764 KB |
testcase_16 | AC | 518 ms
31,892 KB |
testcase_17 | AC | 522 ms
31,892 KB |
testcase_18 | AC | 536 ms
31,760 KB |
testcase_19 | AC | 519 ms
31,764 KB |
testcase_20 | AC | 513 ms
31,764 KB |
testcase_21 | AC | 522 ms
31,760 KB |
testcase_22 | AC | 528 ms
31,652 KB |
testcase_23 | AC | 521 ms
31,896 KB |
testcase_24 | AC | 525 ms
31,888 KB |
testcase_25 | AC | 510 ms
31,896 KB |
testcase_26 | AC | 523 ms
31,892 KB |
testcase_27 | AC | 519 ms
31,892 KB |
testcase_28 | AC | 520 ms
31,764 KB |
testcase_29 | AC | 464 ms
31,868 KB |
testcase_30 | AC | 469 ms
31,892 KB |
testcase_31 | AC | 1 ms
5,376 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; #define DEBUG(x) cerr<<#x<<": "<<x<<endl; #define DEBUG_VEC(v) cerr<<#v<<":";for(int i=0;i<v.size();i++) cerr<<" "<<v[i]; cerr<<endl; #define DEBUG_MAT(v) cerr<<#v<<endl;for(int i=0;i<v.size();i++){for(int j=0;j<v[i].size();j++) {cerr<<v[i][j]<<" ";}cerr<<endl;} typedef long long ll; #define int ll #define vi vector<int> #define vl vector<ll> #define vii vector< vector<int> > #define vll vector< vector<ll> > #define vs vector<string> #define pii pair<int,int> #define pis pair<int,string> #define psi pair<string,int> #define pll pair<ll,ll> template<class S, class T> pair<S, T> operator+(const pair<S, T> &s, const pair<S, T> &t) { return pair<S, T>(s.first + t.first, s.second + t.second); } template<class S, class T> pair<S, T> operator-(const pair<S, T> &s, const pair<S, T> &t) { return pair<S, T>(s.first - t.first, s.second - t.second); } template<class S, class T> ostream& operator<<(ostream& os, pair<S, T> p) { os << "(" << p.first << ", " << p.second << ")"; return os; } #define X first #define Y second #define rep(i,n) for(int i=0;i<(int)(n);i++) #define rep1(i,n) for(int i=1;i<=(int)(n);i++) #define rrep(i,n) for(int i=(int)(n)-1;i>=0;i--) #define rrep1(i,n) for(int i=(int)(n);i>0;i--) #define REP(i,a,b) for(int i=a;i<b;i++) #define in(x, a, b) (a <= x && x < b) #define all(c) c.begin(),c.end() template<class T> bool chmax(T &a, const T &b) { if (a<b) { a = b; return 1; } return 0; } template<class T> bool chmin(T &a, const T &b) { if (a>b) { a = b; return 1; } return 0; } #define UNIQUE(v) v.erase(std::unique(v.begin(), v.end()), v.end()); const ll inf = 1000000001; const ll INF = (ll)1e18 + 1; const long double pi = 3.1415926535897932384626433832795028841971L; #define Sp(p) cout<<setprecision(25)<< fixed<<p<<endl; //int dx[4] = {1, 0, -1, 0}, dy[4] = {0, 1, 0, -1}; //int dx2[8] = { 1,1,0,-1,-1,-1,0,1 }, dy2[8] = { 0,1,1,1,0,-1,-1,-1 }; vi dx = {1, 0, -1, 0}, dy = {0, 1, 0, -1}; vi dx2 = { 1,1,0,-1,-1,-1,0,1 }, dy2 = { 0,1,1,1,0,-1,-1,-1 }; #define fio() cin.tie(0); ios::sync_with_stdio(false); //const ll MOD = 1000000007; const ll MOD = 998244353; // #define mp make_pair //#define endl '\n' // NTT<998244353, 3>で宣言 // グローバルで一度宣言すれば何度でも使用可能 // https://atcoder.jp/contests/jsc2019-final/submissions/7782824 template< int mod, int primitiveroot > struct NTT { vector< vector< int > > rts, rrts; void ensure_base(int N) { if(rts.size() >= N) return; rts.resize(N), rrts.resize(N); for(int i = 1; i < N; i <<= 1) { if(rts[i].size()) continue; int w = mod_pow(primitiveroot, (mod - 1) / (i * 2)); int rw = inverse(w); rts[i].resize(i), rrts[i].resize(i); rts[i][0] = 1, rrts[i][0] = 1; for(int k = 1; k < i; k++) { rts[i][k] = mul(rts[i][k - 1], w); rrts[i][k] = mul(rrts[i][k - 1], rw); } } } inline int mod_pow(int x, int n) { int ret = 1; while(n > 0) { if(n & 1) ret = mul(ret, x); x = mul(x, x); n >>= 1; } return ret; } inline int inverse(int x) { return mod_pow(x, mod - 2); } inline int add(int x, int y) { x += y; if(x >= mod) x -= mod; return x; } inline int mul(int a, int b) { return int(1LL * a * b % mod); } void DiscreteFourierTransform(vector< int > &F, bool rev) { const int N = (int) F.size(); ensure_base(N); for(int i = 0, j = 1; j + 1 < N; j++) { for(int k = N >> 1; k > (i ^= k); k >>= 1); if(i > j) swap(F[i], F[j]); } for(int i = 1; i < N; i <<= 1) { for(int j = 0; j < N; j += i * 2) { for(int k = 0; k < i; k++) { int s = F[j + k], t = mul(F[j + k + i], rev ? rrts[i][k] : rts[i][k]); F[j + k] = add(s, t), F[j + k + i] = add(s, mod - t); } } } if(rev) { int temp = inverse(N); for(int i = 0; i < N; i++) F[i] = mul(F[i], temp); } } vector< int > Multiply(const vector< int > &A, const vector< int > &B) { int sz = 1; while(sz < A.size() + B.size() - 1) sz <<= 1; vector< int > F(sz), G(sz); for(int i = 0; i < A.size(); i++) F[i] = A[i]; for(int i = 0; i < B.size(); i++) G[i] = B[i]; DiscreteFourierTransform(F, false); DiscreteFourierTransform(G, false); for(int i = 0; i < sz; i++) F[i] = mul(F[i], G[i]); DiscreteFourierTransform(F, true); F.resize(A.size() + B.size() - 1); return F; } }; NTT<998244353, 3> ntt; vl calc(vl& a, int l, int r) { if (l + 1 == r) { vl res = {a[l] - 1, 1}; return res; } int m = (l + r) / 2; vl b1 = calc(a, l, m), b2 = calc(a, m, r); vl res = ntt.Multiply(b1, b2); return res; } signed main() { int n, q; cin >> n >> q; vl a(n); rep (i, n) { cin >> a[i]; a[i] %= MOD; } auto res = calc(a, 0, n); while (q--) { int x; cin >> x; cout << res[x] << endl; } }