結果
問題 | No.1100 Boxes |
ユーザー |
![]() |
提出日時 | 2020-06-26 22:04:17 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 56 ms / 2,000 ms |
コード長 | 18,974 bytes |
コンパイル時間 | 3,039 ms |
コンパイル使用メモリ | 229,100 KB |
最終ジャッジ日時 | 2025-01-11 11:29:44 |
ジャッジサーバーID (参考情報) |
judge3 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 36 |
ソースコード
#pragma GCC optimize ("Ofast")#include<bits/stdc++.h>using namespace std;#define MD (998244353U)#define MD_PRIMITIVE_ROOT (3U)#define PI 3.14159265358979323846void *wmem;char memarr[96000000];template<class S, class T> inline S max_L(S a,T b){return a>=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);}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+=(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;}}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 T, class S> inline T pow_L(T a, S b){T res = 1;res = 1;for(;;){if(b&1){res *= a;}b >>= 1;if(b==0){break;}a *= a;}return res;}inline double pow_L(double a, double b){return pow(a,b);}template<class S, class T> inline S chmax(S &a, T b){if(a<b){a=b;}return a;}template<class T> struct Comb{int mem_fact;T *factri;T *ifactri;Comb(){mem_fact = 0;}inline void expand_fact(int k){if(k <= mem_fact){return;}chmax(k, 2* mem_fact);if(mem_fact == 0){int i;factri = (T*)malloc(k * sizeof(T));ifactri = (T*)malloc(k * sizeof(T));factri[0] = 1;for(i=(1);i<(k);i++){factri[i] = i * factri[i-1];}ifactri[k-1] = 1 / factri[k-1];for(i=(k-1)-1;i>=(0);i--){ifactri[i] = (i+1) * ifactri[i+1];}}else{int i;factri = (T*)realloc(factri, k * sizeof(T));ifactri = (T*)realloc(ifactri, k * sizeof(T));for(i=(mem_fact);i<(k);i++){factri[i] = i * factri[i-1];}ifactri[k-1] = 1 / factri[k-1];for(i=(k-1)-1;i>=(mem_fact);i--){ifactri[i] = (i+1) * ifactri[i+1];}}mem_fact = k;}inline T fac(int k){if(mem_fact < k+1){expand_fact(k+1);}return factri[k];}inline T ifac(int k){if(mem_fact < k+1){expand_fact(k+1);}return ifactri[k];}inline T C(int a, int b){if(b < 0 || b > a){return 0;}if(mem_fact < a+1){expand_fact(a+1);}return factri[a] * ifactri[b] * ifactri[a-b];}inline T P(int a, int b){if(b < 0 || b > a){return 0;}if(mem_fact < a+1){expand_fact(a+1);}return factri[a] * ifactri[a-b];}inline T H(int a, int b){if(a==0 && b==0){return 1;}if(a <= 0 || b < 0){return 0;}if(mem_fact < a+b){expand_fact(a+b);}return C(a+b-1, b);}inline T Multinomial(int sz, int a[]){int i;int s = 0;T res;for(i=(0);i<(sz);i++){s += a[i];}if(mem_fact < s+1){expand_fact(s+1);}res = factri[s];for(i=(0);i<(sz);i++){res *= ifactri[a[i]];}return 1;}inline T Multinomial(int a){return 1;}inline T Multinomial(int a, int b){if(mem_fact < a+b+1){expand_fact(a+b+1);}return factri[a+b] * ifactri[a] * ifactri[b];}inline T Multinomial(int a, int b, int c){if(mem_fact < a+b+c+1){expand_fact(a+b+c+1);}return factri[a+b+c] * ifactri[a] * ifactri[b] * ifactri[c];}inline T Multinomial(int a, int b, int c, int d){if(mem_fact < a+b+c+d+1){expand_fact(a+b+c+d+1);}return factri[a+b+c+d] * ifactri[a] * ifactri[b] * ifactri[c] * ifactri[d];}inline T Catalan(int n){if(n < 0){return 0;}if(mem_fact < 2*n+1){expand_fact(2*n+1);}return factri[2*n] * ifactri[n] * ifactri[n+1];}inline T C_s(long long a, long long b){long long i;T res;if(b < 0 || b > a){return 0;}if(b > a - b){b = a - b;}res = 1;for(i=(0);i<(b);i++){res *= a - i;res /= i + 1;}return res;}inline T P_s(long long a, long long b){long long i;T res;if(b < 0 || b > a){return 0;}res = 1;for(i=(0);i<(b);i++){res *= a - i;}return res;}inline T per_s(long long n, long long k){T d;int m;if(n < 0 || k < 0){return 0;}if(n == k && k == 0){return 1;}if(n == 0 || k == 0){return 0;}if(k==1){return 1;}if(k==2){d = n / 2;return d;}if(k==3){d = (n-1) / 6;m = (n-1) % 6;if(m==0){return 3 * d * d + d;}if(m==1){return 3 * d * d + 2 * d;}if(m==2){return 3 * d * d + 3 * d + 1;}if(m==3){return 3 * d * d + 4 * d + 1;}if(m==4){return 3 * d * d + 5 * d + 2;}if(m==5){return 3 * d * d + 6 * d + 3;}}assert(0 && "per_s should be k <= 3");return -1;}};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_L(int n, fft_pnt x[], void *mem){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_L(int n, fft_pnt x[], void *mem){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 jG1yfsum = n2;for(i=(As);i<(jG1yfsum);i++){a[i].set(0,0);}for(i=(0);i<(Bs);i++){b[i].set(B[i], 0);}int grBCmONb = n2;for(i=(Bs);i<(grBCmONb);i++){b[i].set(0,0);}fft_L(n2, a, mem);fft_L(n2, b, mem);for(i=(0);i<(n2);i++){a[i] *= b[i];}fftinv_L(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 memzHCbB = n2;for(i=(As);i<(memzHCbB);i++){a[i].set(0,0);}fft_L(n2, a, mem);for(i=(0);i<(n2);i++){a[i] *= a[i];}fftinv_L(n2, a, mem);mul = 1.0 / n2;for(i=(0);i<(Rs);i++){res[i] = a[i].x * mul;}}void fft_L(int n, Modint x[], Modint root, void *mem){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_L(int n, Modint x[], Modint root, void *mem){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 rv4PtiIc = n2;for(i=(As);i<(rv4PtiIc);i++){a[i].val = 0;}for(i=(0);i<(Bs);i++){b[i] = B[i];}int emd5LSgV = n2;for(i=(Bs);i<(emd5LSgV);i++){b[i].val = 0;}fft_L(n2, a, root, mem);fft_L(n2, b, root, mem);for(i=(0);i<(n2);i++){a[i] *= b[i];}fftinv_L(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 IyTJmJ0I = n2;for(i=(As);i<(IyTJmJ0I);i++){a[i].val = 0;}fft_L(n2, a, root, mem);for(i=(0);i<(n2);i++){a[i] *= a[i];}fftinv_L(n2, a, root, mem);r = Modint(n2).inverse();for(i=(0);i<(Rs);i++){res[i] = a[i]*r;}}int N;int K;Modint x[100000+2];Modint y[100000+2];Modint z[100000+2];int main(){int i;wmem = memarr;Modint res = 0;Comb<Modint> c;rd(N);rd(K);for(i=(0);i<(100000+2);i++){x[i] = ((pow_L((Modint(i)),N))) * c.ifac(i);y[i] = c.ifac(i);if(i%2){y[i] = -y[i];}}convolution_L(x,100000+2,y,100000+2,z,100000+2);for(i=(0);i<(K+1);i++){if((K-i)%2){res += c.C(K,i) * z[i] * c.fac(i);}}wt_L(res);wt_L('\n');return 0;}// cLay varsion 20200509-1// --- original code ---// #define MD 998244353// int N, K;// Modint x[1d5+2], y[1d5+2], z[1d5+2];// {// Modint res = 0;// Comb<Modint> c;// rd(N,K);// rep(i,1d5+2){// x[i] = ((Modint(i)) ** N) * c.ifac(i);// y[i] = c.ifac(i);// if(i%2) y[i] = -y[i];// }// convolution(x,1d5+2,y,1d5+2,z,1d5+2);// rep(i,K+1) if((K-i)%2) res += c.C(K,i) * z[i] * c.fac(i);// wt(res);// }