結果
問題 | No.1549 [Cherry 2nd Tune] BANning Tuple |
ユーザー |
![]() |
提出日時 | 2021-06-11 22:17:51 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
TLE
|
実行時間 | - |
コード長 | 18,015 bytes |
コンパイル時間 | 3,702 ms |
コンパイル使用メモリ | 248,016 KB |
最終ジャッジ日時 | 2025-01-22 06:06:46 |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 11 TLE * 9 |
ソースコード
#pragma GCC optimize("Ofast")#pragma GCC optimize("unroll-loops")#pragma GCC optimize("inline")#include<bits/stdc++.h>using namespace std;#define MD (998244353U)#define MD_PRIMITIVE_ROOT (3U)#define PI 3.14159265358979323846template<class T> struct cLtraits_identity{using type = T;};template<class T> using cLtraits_try_make_signed =typename conditional<is_integral<T>::value,make_signed<T>,cLtraits_identity<T>>::type;template <class S, class T> struct cLtraits_common_type{using tS = typename cLtraits_try_make_signed<S>::type;using tT = typename cLtraits_try_make_signed<T>::type;using type = typename common_type<tS,tT>::type;};void*wmem;char memarr[96000000];template<class S, class T> inline auto max_L(S a, T b)-> typename cLtraits_common_type<S,T>::type{return (typename cLtraits_common_type<S,T>::type) a >= (typename cLtraits_common_type<S,T>::type) b ? a : b;}template<class T> inline void walloc1d(T **arr, int x, void **mem = &wmem){static int skip[16] = {0, 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1};(*mem) = (void*)( ((char*)(*mem)) + skip[((unsigned long long)(*mem)) & 15] );(*arr)=(T*)(*mem);(*mem)=((*arr)+x);}template<class T> inline void walloc1d(T **arr, int x1, int x2, void **mem = &wmem){walloc1d(arr, x2-x1, mem);(*arr) -= x1;}struct Modint{unsigned val;Modint(){val=0;}Modint(int a){val = ord(a);}Modint(unsigned a){val = ord(a);}Modint(long long a){val = ord(a);}Modint(unsigned long long a){val = ord(a);}inline unsigned ord(unsigned a){return a%MD;}inline unsigned ord(int a){a %= (int)MD;if(a < 0){a += MD;}return a;}inline unsigned ord(unsigned long long a){return a%MD;}inline unsigned ord(long long a){a %= (int)MD;if(a < 0){a += MD;}return a;}inline unsigned get(){return val;}inline Modint &operator++(){val++;if(val >= MD){val -= MD;}return *this;}inline Modint &operator--(){if(val == 0){val = MD - 1;}else{--val;}return *this;}inline Modint operator++(int a){Modint res(*this);val++;if(val >= MD){val -= MD;}return res;}inline Modint operator--(int a){Modint res(*this);if(val == 0){val = MD - 1;}else{--val;}return res;}inline Modint &operator+=(Modint a){val += a.val;if(val >= MD){val -= MD;}return *this;}inline Modint &operator-=(Modint a){if(val < a.val){val = val + MD - a.val;}else{val -= a.val;}return *this;}inline Modint &operator*=(Modint a){val = ((unsigned long long)val*a.val)%MD;return *this;}inline Modint &operator/=(Modint a){return *this *= a.inverse();}inline Modint operator+(Modint a){return Modint(*this)+=a;}inline Modint operator-(Modint a){return Modint(*this)-=a;}inline Modint operator*(Modint a){return Modint(*this)*=a;}inline Modint operator/(Modint a){return Modint(*this)/=a;}inline Modint operator+(int a){return Modint(*this)+=Modint(a);}inline Modint operator-(int a){return Modint(*this)-=Modint(a);}inline Modint operator*(int a){return Modint(*this)*=Modint(a);}inline Modint operator/(int a){return Modint(*this)/=Modint(a);}inline Modint operator+(long long a){return Modint(*this)+=Modint(a);}inline Modint operator-(long long a){return Modint(*this)-=Modint(a);}inline Modint operator*(long long a){return Modint(*this)*=Modint(a);}inline Modint operator/(long long a){return Modint(*this)/=Modint(a);}inline Modint operator-(void){Modint res;if(val){res.val=MD-val;}else{res.val=0;}return res;}inline operator bool(void){return val!=0;}inline operator int(void){return get();}inline operator long long(void){return get();}inline Modint inverse(){int a = val;int b = MD;int u = 1;int v = 0;int t;Modint res;while(b){t = a / b;a -= t * b;swap(a, b);u -= t * v;swap(u, v);}if(u < 0){u += MD;}res.val = u;return res;}inline Modint pw(unsigned long long b){Modint a(*this);Modint res;res.val = 1;while(b){if(b&1){res *= a;}b >>= 1;a *= a;}return res;}inline bool operator==(int a){return ord(a)==val;}inline bool operator!=(int a){return ord(a)!=val;}};inline Modint operator+(int a, Modint b){return Modint(a)+=b;}inline Modint operator-(int a, Modint b){return Modint(a)-=b;}inline Modint operator*(int a, Modint b){return Modint(a)*=b;}inline Modint operator/(int a, Modint b){return Modint(a)/=b;}inline Modint operator+(long long a, Modint b){return Modint(a)+=b;}inline Modint operator-(long long a, Modint b){return Modint(a)-=b;}inline Modint operator*(long long a, Modint b){return Modint(a)*=b;}inline Modint operator/(long long a, Modint b){return Modint(a)/=b;}inline int my_getchar_unlocked(){static char buf[1048576];static int s = 1048576;static int e = 1048576;if(s == e && e == 1048576){e = fread_unlocked(buf, 1, 1048576, stdin);s = 0;}if(s == e){return EOF;}return buf[s++];}inline void rd(int &x){int k;int m=0;x=0;for(;;){k = my_getchar_unlocked();if(k=='-'){m=1;break;}if('0'<=k&&k<='9'){x=k-'0';break;}}for(;;){k = my_getchar_unlocked();if(k<'0'||k>'9'){break;}x=x*10+k-'0';}if(m){x=-x;}}inline void rd(long long &x){int k;int m=0;x=0;for(;;){k = my_getchar_unlocked();if(k=='-'){m=1;break;}if('0'<=k&&k<='9'){x=k-'0';break;}}for(;;){k = my_getchar_unlocked();if(k<'0'||k>'9'){break;}x=x*10+k-'0';}if(m){x=-x;}}struct MY_WRITER{char buf[1048576];int s;int e;MY_WRITER(){s = 0;e = 1048576;}~MY_WRITER(){if(s){fwrite_unlocked(buf, 1, s, stdout);}}};MY_WRITER MY_WRITER_VAR;void my_putchar_unlocked(int a){if(MY_WRITER_VAR.s == MY_WRITER_VAR.e){fwrite_unlocked(MY_WRITER_VAR.buf, 1, MY_WRITER_VAR.s, stdout);MY_WRITER_VAR.s = 0;}MY_WRITER_VAR.buf[MY_WRITER_VAR.s++] = a;}inline void wt_L(char a){my_putchar_unlocked(a);}inline void wt_L(int x){int s=0;int m=0;char f[10];if(x<0){m=1;x=-x;}while(x){f[s++]=x%10;x/=10;}if(!s){f[s++]=0;}if(m){my_putchar_unlocked('-');}while(s--){my_putchar_unlocked(f[s]+'0');}}inline void wt_L(Modint x){int i;i = (int)x;wt_L(i);}template<class S, class T> inline S chmax(S &a, T b){if(a<b){a=b;}return a;}struct fft_pnt{double x;double y;fft_pnt(void){}fft_pnt(double a, double b){x = a;y = b;}void set(double a, double b){x = a;y = b;}fft_pnt& operator+=(fft_pnt a){x+=a.x;y+=a.y;return *this;}fft_pnt& operator-=(fft_pnt a){x-=a.x;y-=a.y;return *this;}fft_pnt& operator*=(fft_pnt a){fft_pnt p = *this;x = p.x*a.x-p.y*a.y;y = p.x*a.y+p.y*a.x;return *this;}fft_pnt operator+(fft_pnt a){return fft_pnt(*this) += a;}fft_pnt operator-(fft_pnt a){return fft_pnt(*this) -= a;}fft_pnt operator*(fft_pnt a){return fft_pnt(*this) *= a;}};void fft(int n, fft_pnt x[], void *mem = wmem){int i;int j;int n1;int n2;int n3;int step = 1;double theta = 2*PI / n;double tmp;fft_pnt w1;fft_pnt w2;fft_pnt w3;fft_pnt a;fft_pnt b;fft_pnt c;fft_pnt d;fft_pnt aa;fft_pnt bb;fft_pnt cc;fft_pnt dd;fft_pnt*y = (fft_pnt*)mem;while(n > 2){n1 = n / 4;n2 = n1 + n1;n3 = n1 + n2;for(i=(0);i<(n1);i++){w1 = fft_pnt(cos(i*theta),-sin(i*theta));w2 = w1*w1;w3 = w1*w2;for(j=(0);j<(step);j++){a = x[j+step*i];b = x[j+step*(i+n1)];c = x[j+step*(i+n2)];d = x[j+step*(i+n3)];aa = a + c;bb = a - c;cc = b + d;dd = b - d;tmp = dd.y;dd.y = dd.x;dd.x = -tmp;y[j+step*(4*i )] = aa + cc;y[j+step*(4*i+1)] = w1*(bb - dd);y[j+step*(4*i+2)] = w2*(aa - cc);y[j+step*(4*i+3)] = w3*(bb + dd);}}n /= 4;step *= 4;theta *= 4;swap(x,y);}if(n==2){for(i=(0);i<(step);i++){y[i] = x[i] + x[i+step];y[i+step] = x[i] - x[i+step];}n /= 2;step *= 2;theta *= 2;swap(x,y);}for(i=(0);i<(step);i++){y[i] = x[i];}}void fftinv(int n, fft_pnt x[], void *mem = wmem){int i;int j;int n1;int n2;int n3;int step = 1;double theta = 2*PI / n;double tmp;fft_pnt w1;fft_pnt w2;fft_pnt w3;fft_pnt a;fft_pnt b;fft_pnt c;fft_pnt d;fft_pnt aa;fft_pnt bb;fft_pnt cc;fft_pnt dd;fft_pnt*y = (fft_pnt*)mem;while(n > 2){n1 = n / 4;n2 = n1 + n1;n3 = n1 + n2;for(i=(0);i<(n1);i++){w1 = fft_pnt(cos(i*theta),sin(i*theta));w2 = w1*w1;w3 = w1*w2;for(j=(0);j<(step);j++){a = x[j+step*i];b = x[j+step*(i+n1)];c = x[j+step*(i+n2)];d = x[j+step*(i+n3)];aa = a + c;bb = a - c;cc = b + d;dd = b - d;tmp = dd.y;dd.y = dd.x;dd.x = -tmp;y[j+step*(4*i )] = aa + cc;y[j+step*(4*i+1)] = w1*(bb + dd);y[j+step*(4*i+2)] = w2*(aa - cc);y[j+step*(4*i+3)] = w3*(bb - dd);}}n /= 4;step *= 4;theta *= 4;swap(x,y);}if(n==2){for(i=(0);i<(step);i++){y[i] = x[i] + x[i+step];y[i+step] = x[i] - x[i+step];}n /= 2;step *= 2;theta *= 2;swap(x,y);}for(i=(0);i<(step);i++){y[i] = x[i];}}void convolution_L(double A[], int As, double B[], int Bs, double res[], int Rs, void *mem = wmem){int i;int n;int n2;double mul;fft_pnt*a;fft_pnt*b;n =max_L(As+Bs, Rs);for(n2=1;n2<n;n2*=2){;}walloc1d(&a, n2, &mem);walloc1d(&b, n2, &mem);for(i=(0);i<(As);i++){a[i].set(A[i], 0);}int grBCmONb = n2;for(i=(As);i<(grBCmONb);i++){a[i].set(0,0);}for(i=(0);i<(Bs);i++){b[i].set(B[i], 0);}int eNrGll8F = n2;for(i=(Bs);i<(eNrGll8F);i++){b[i].set(0,0);}fft(n2, a, mem);fft(n2, b, mem);for(i=(0);i<(n2);i++){a[i] *= b[i];}fftinv(n2, a, mem);mul = 1.0 / n2;for(i=(0);i<(Rs);i++){res[i] = a[i].x * mul;}}void convolution_L(double A[], int As, double res[], int Rs, void *mem = wmem){int i;int n;int n2;double mul;fft_pnt*a;n =max_L(As+As, Rs);for(n2=1;n2<n;n2*=2){;}walloc1d(&a, n2, &mem);for(i=(0);i<(As);i++){a[i].set(A[i], 0);}int niykphg5 = n2;for(i=(As);i<(niykphg5);i++){a[i].set(0,0);}fft(n2, a, mem);for(i=(0);i<(n2);i++){a[i] *= a[i];}fftinv(n2, a, mem);mul = 1.0 / n2;for(i=(0);i<(Rs);i++){res[i] = a[i].x * mul;}}void fft(int n, Modint x[], Modint root = MD_PRIMITIVE_ROOT, void *mem = wmem){int i;int j;int n1;int n2;int n3;int step = 1;Modint w1;Modint w2;Modint w3;Modint a;Modint b;Modint c;Modint d;Modint aa;Modint bb;Modint cc;Modint dd;Modint tmp;Modint*y;walloc1d(&y, n, &mem);tmp = root.pw((MD-1)/4*3);root = root.pw((MD-1)/n);while(n > 2){n1 = n / 4;n2 = n1 + n1;n3 = n1 + n2;w1.val = 1;for(i=(0);i<(n1);i++){w2 = w1*w1;w3 = w1*w2;for(j=(0);j<(step);j++){a = x[j+step*i];b = x[j+step*(i+n1)];c = x[j+step*(i+n2)];d = x[j+step*(i+n3)];aa = a + c;bb = a - c;cc = b + d;dd = (b - d) * tmp;y[j+step*(4*i )] = aa + cc;y[j+step*(4*i+1)] = w1*(bb - dd);y[j+step*(4*i+2)] = w2*(aa - cc);y[j+step*(4*i+3)] = w3*(bb + dd);}w1 *= root;}n /= 4;step *= 4;root *= root;root *= root;swap(x,y);}if(n==2){for(i=(0);i<(step);i++){y[i] = x[i] + x[i+step];y[i+step] = x[i] - x[i+step];}n /= 2;step *= 2;root *= root;swap(x,y);}for(i=(0);i<(step);i++){y[i] = x[i];}}void fftinv(int n, Modint x[], Modint root = MD_PRIMITIVE_ROOT, void *mem = wmem){int i;int j;int n1;int n2;int n3;int step = 1;Modint w1;Modint w2;Modint w3;Modint a;Modint b;Modint c;Modint d;Modint aa;Modint bb;Modint cc;Modint dd;Modint tmp;Modint*y;walloc1d(&y, n, &mem);root = root.inverse();tmp = root.pw((MD-1)/4);root = root.pw((MD-1)/n);while(n > 2){n1 = n / 4;n2 = n1 + n1;n3 = n1 + n2;w1.val = 1;for(i=(0);i<(n1);i++){w2 = w1*w1;w3 = w1*w2;for(j=(0);j<(step);j++){a = x[j+step*i];b = x[j+step*(i+n1)];c = x[j+step*(i+n2)];d = x[j+step*(i+n3)];aa = a + c;bb = a - c;cc = b + d;dd = (b - d) * tmp;y[j+step*(4*i )] = aa + cc;y[j+step*(4*i+1)] = w1*(bb + dd);y[j+step*(4*i+2)] = w2*(aa - cc);y[j+step*(4*i+3)] = w3*(bb - dd);}w1 *= root;}n /= 4;step *= 4;root *= root;root *= root;swap(x,y);}if(n==2){for(i=(0);i<(step);i++){y[i] = x[i] + x[i+step];y[i+step] = x[i] - x[i+step];}n /= 2;step *= 2;root *= root;swap(x,y);}for(i=(0);i<(step);i++){y[i] = x[i];}}void convolution_L(Modint A[], int As, Modint B[], int Bs, Modint res[], int Rs, Modint root = MD_PRIMITIVE_ROOT, void *mem = wmem){int i;int n;int n2;Modint*a;Modint*b;Modint r;n =max_L(As+Bs, Rs);for(n2=1;n2<n;n2*=2){;}walloc1d(&a, n2, &mem);walloc1d(&b, n2, &mem);for(i=(0);i<(As);i++){a[i] = A[i];}int emd5LSgV = n2;for(i=(As);i<(emd5LSgV);i++){a[i].val = 0;}for(i=(0);i<(Bs);i++){b[i] = B[i];}int jIDgiLP1 = n2;for(i=(Bs);i<(jIDgiLP1);i++){b[i].val = 0;}fft(n2, a, root, mem);fft(n2, b, root, mem);for(i=(0);i<(n2);i++){a[i] *= b[i];}fftinv(n2, a, root, mem);r = Modint(n2).inverse();for(i=(0);i<(Rs);i++){res[i] = a[i] * r;}}void convolution_L(Modint A[], int As, Modint res[], int Rs, Modint root = MD_PRIMITIVE_ROOT, void *mem = wmem){int i;int n;int n2;Modint*a;Modint r;n =max_L(2*As, Rs);for(n2=1;n2<n;n2*=2){;}walloc1d(&a, n2, &mem);for(i=(0);i<(As);i++){a[i] = A[i];}int AUdoqz3N = n2;for(i=(As);i<(AUdoqz3N);i++){a[i].val = 0;}fft(n2, a, root, mem);for(i=(0);i<(n2);i++){a[i] *= a[i];}fftinv(n2, a, root, mem);r = Modint(n2).inverse();for(i=(0);i<(Rs);i++){res[i] = a[i]*r;}}long long N;long long K[100];int Q;int A[100];int B[100];int S[100];int T[100];Modint ok[100][3001];Modint dp[3001];Modint nx[3001];map<long long,int> ind;Modint H(long long a, int b){int i;Modint x;Modint y;if(a == b && b == 0){return 1;}if(a <= 0 || b < 0){return 0;}x = y = 1;for(i=(0);i<(b);i++){x *= a+i;y *= i+1;}return x / y;}int main(){int i, q;wmem = memarr;int mx = -1;int sz = 0;Modint res;rd(N);rd(Q);{int Q5VJL1cS;for(Q5VJL1cS=(0);Q5VJL1cS<(Q);Q5VJL1cS++){rd(K[Q5VJL1cS]);rd(A[Q5VJL1cS]);rd(B[Q5VJL1cS]);rd(S[Q5VJL1cS]);rd(T[Q5VJL1cS]);}}for(i=(0);i<(Q);i++){int j;for(j=(0);j<(3001);j++){ok[i][j] = 1;}}for(i=(0);i<(Q);i++){if(ind.count(K[i])){K[i] = ind[K[i]];continue;}K[i] = ind[K[i]] = sz++;}for(q=(0);q<(Q);q++){chmax(mx, K[q]+1);for(i=(A[q]);i<(B[q]+1);i++){ok[K[q]][i] = 0;}for(i=(0);i<(3001);i++){dp[i] = ok[0][i];}for(i=(1);i<(mx);i++){int j;convolution_L(dp, 3001, ok[i], 3001, nx, 3001);for(j=(0);j<(3001);j++){dp[j] = nx[j];}}res = 0;for(i=(0);i<(T[q]+1);i++){if(dp[i]){res += H(N-mx+1, T[q]-i) * dp[i];res -= H(N-mx+1, S[q]-i-1) * dp[i];}}wt_L(res);wt_L('\n');}return 0;}// cLay version 20210611-1 [beta]// --- original code ---// #define MD 998244353// ll N, K[100];// int Q, A[], B[], S[], T[];// Modint ok[100][3001];// Modint dp[3001], nx[3001];// map<ll,int> ind;//// Modint H(ll a, int b){// int i;// Modint x, y;// if(a == b == 0) return 1;// if(a <= 0 || b < 0) return 0;// x = y = 1;// rep(i,b){// x *= a+i;// y *= i+1;// }// return x / y;// }//// {// int mx = -1, sz = 0;// Modint res;// rd(N,Q,(K,A,B,S,T)(Q));// rep(i,Q) rep(j,3001) ok[i][j] = 1;//// rep(i,Q){// if(ind.count(K[i])) K[i] = ind[K[i]], continue;// K[i] = ind[K[i]] = sz++;// }//// rep(q,Q){// mx >?= K[q]+1;// rep(i,A[q],B[q]+1) ok[K[q]][i] = 0;// rep(i,3001) dp[i] = ok[0][i];// rep(i,1,mx){// convolution(dp, 3001, ok[i], 3001, nx, 3001);// rep(j,3001) dp[j] = nx[j];// }// res = 0;// rep(i,T[q]+1) if(dp[i]){// res += H(N-mx+1, T[q]-i) * dp[i];// res -= H(N-mx+1, S[q]-i-1) * dp[i];// }// wt(res);// }// }