QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#416601#8715. 放苹果forget-starAC ✓593ms42864kbC++143.4kb2024-05-21 23:52:512024-05-21 23:52:51

Judging History

This is the latest submission verdict.

  • [2024-05-21 23:52:51]
  • Judged
  • Verdict: AC
  • Time: 593ms
  • Memory: 42864kb
  • [2024-05-21 23:52:51]
  • Submitted

answer

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<map>
#include<vector>
#include<queue>
#include<set>
#include<bitset>
#include<deque>
#include<stack>
#include<unordered_map>
#include<ctime>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
const int N=4e5+10,MOD=998244353;
int n,m,f[N<<2],tmp[N<<2],r[N<<2],g[N<<2],C[N<<2],D[N<<2];
int ksm(int x,int y)
{
	int res=1;
	while(y>0)
	{
		if(y&1) res=1ll*res*x%MOD;
		y>>=1;x=1ll*x*x%MOD;
	}
	return res;
}
int get_r(int len)
{
	int tt=0,mi=1;while(mi<=len) mi<<=1,tt++;tt--;
	for(int i=0;i<mi;i++) r[i]=(r[i>>1]>>1)|((i&1)<<tt);
	return mi;
}
void NTT(int *f,int len,int typ)
{
	for(int i=0;i<len;i++) if(i<r[i]) swap(f[i],f[r[i]]);
	for(int i=1;i<len;i<<=1)
	{
		int w=ksm(3,(MOD-1)/(i<<1));
		if(typ==-1) w=ksm(w,MOD-2);
		for(int j=0,p=i<<1;j<len;j+=p)
			for(int k=0,t=1;k<i;k++,t=1ll*t*w%MOD)
			{
				int x=f[j+k],y=1ll*f[i+j+k]*t%MOD;
				f[j+k]=(0ll+x+y)%MOD;
				f[i+j+k]=(0ll+x-y+MOD)%MOD;
			}
	}
	if(typ==-1)
	{
		int inv=ksm(len,MOD-2);
		for(int i=0;i<len;i++) f[i]=1ll*f[i]*inv%MOD;
	}
	return;
}
void get_inv(int *f,int *g,int len)
{
    if(len==1){g[0]=ksm(f[0],MOD-2);return;}
    get_inv(f,g,len>>1);
    int mi=get_r(2*len);
    for(int i=0;i<len;i++) C[i]=f[i],D[i]=g[i];
    for(int i=len;i<mi;i++) C[i]=D[i]=0;
    NTT(C,mi,1);NTT(D,mi,1);
    for(int i=0;i<mi;i++) D[i]=1ll*(2ll-1ll*C[i]*D[i]%MOD+MOD)*D[i]%MOD;
    NTT(D,mi,-1);
    for(int i=0;i<len;i++) g[i]=D[i];
    for(int i=len;i<mi;i++) g[i]=0;
    return;
}
int fac[N],inv[N],p[N<<2];
int main()
{
	scanf("%d%d",&n,&m);
	
	fac[0]=1;
	for(int i=1;i<=n+5;i++) fac[i]=1ll*i*fac[i-1]%MOD;
	inv[n+5]=ksm(fac[n+5],MOD-2);
	for(int i=n+5;i>=1;i--) inv[i-1]=1ll*i*inv[i]%MOD;
	
//	int ans0=0;
//	for(int j=0;j<=n;j++)
//		for(int k=0;k<=n-j;k++)
//			for(int i=1;i<m;i++)
//				ans0=(0ll+ans0+1ll*min(j,n-j)*fac[n]%MOD*inv[j]%MOD*inv[n-j]%MOD
//				*fac[n-j]%MOD*inv[n-j-k]%MOD*inv[k]%MOD*ksm(i,j+k)%MOD*ksm(m,n-j-k)%MOD*((k&1)?MOD-1:1))%MOD;
//	printf("ans0=%d\n",ans0); 
	
	for(int i=1;i<=n+1;i++) f[i-1]=inv[i];
	int l=1;while(l<=(n+3))l<<=1;
	get_inv(f,g,l);
	memset(f,0,sizeof(f));
	for(int i=1;i<=n+1;i++)
		f[i-1]=(0ll+1ll*inv[i]*ksm(m,i)%MOD-inv[i]+MOD)%MOD;
	int op=get_r(l);
	NTT(f,op,1);NTT(g,op,1);
	for(int i=0;i<op;i++) f[i]=1ll*f[i]*g[i]%MOD;
	NTT(f,op,-1);
	for(int i=0;i<=n;i++) f[i]=1ll*f[i]*fac[i]%MOD;
//	for(int i=0;i<=n;i++) printf("f[%d]=%d ",i,f[i]);puts("");
//	
//	for(int i=0;i<=n;i++)
//	{
//		for(int j=1;j<m;j++) p[i]=(0ll+p[i]+1ll*ksm(j,i))%MOD;
//		printf("p[%d]=%d ",i,p[i]);
//	}
//	puts("");
	
	for(int i=0;i<=n;i++) tmp[i]=f[i];memset(f,0,sizeof(f));
	for(int i=0;i<=n;i++) f[i]=1ll*tmp[n-i]*inv[i]%MOD*ksm(m,i)%MOD;
	memset(g,0,sizeof(g));
	for(int i=0;i<=n;i++) g[i]=1ll*((i&1)?MOD-1:1)*inv[i]%MOD;
	op=get_r(l);
	NTT(f,op,1);NTT(g,op,1);
	for(int i=0;i<op;i++) f[i]=1ll*f[i]*g[i]%MOD;
	NTT(f,op,-1);
//	for(int i=0;i<=n;i++) printf("f[%d]=%d ",i,f[i]);puts("");
//	
//	for(int i=0;i<=n;i++)
//	{
//		int res=0;
//		for(int j=1;j<m;j++) res=(0ll+res+1ll*ksm(j,i)*ksm(m-j,n-i))%MOD;
//		printf("%d ",res);
//	}
//	puts("");
	
	int ans=0;
	for(int i=0;i<=n;i++)
		ans=(0ll+ans+1ll*min(i,n-i)*fac[n]%MOD*inv[i]%MOD*f[n-i])%MOD;
	printf("%d\n",ans);
	return 0;
}

这程序好像有点Bug,我给组数据试试?

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 2ms
memory: 27428kb

input:

2 3

output:

8

result:

ok 1 number(s): "8"

Test #2:

score: 0
Accepted
time: 0ms
memory: 27504kb

input:

3 3

output:

36

result:

ok 1 number(s): "36"

Test #3:

score: 0
Accepted
time: 0ms
memory: 25164kb

input:

1 1

output:

0

result:

ok 1 number(s): "0"

Test #4:

score: 0
Accepted
time: 0ms
memory: 28684kb

input:

1 2

output:

0

result:

ok 1 number(s): "0"

Test #5:

score: 0
Accepted
time: 0ms
memory: 28908kb

input:

1 3

output:

0

result:

ok 1 number(s): "0"

Test #6:

score: 0
Accepted
time: 0ms
memory: 29504kb

input:

2 1

output:

0

result:

ok 1 number(s): "0"

Test #7:

score: 0
Accepted
time: 0ms
memory: 26924kb

input:

3 1

output:

0

result:

ok 1 number(s): "0"

Test #8:

score: 0
Accepted
time: 12ms
memory: 29188kb

input:

3719 101

output:

78994090

result:

ok 1 number(s): "78994090"

Test #9:

score: 0
Accepted
time: 7ms
memory: 29356kb

input:

2189 1022

output:

149789741

result:

ok 1 number(s): "149789741"

Test #10:

score: 0
Accepted
time: 3ms
memory: 28608kb

input:

2910 382012013

output:

926541722

result:

ok 1 number(s): "926541722"

Test #11:

score: 0
Accepted
time: 575ms
memory: 39232kb

input:

131072 3837829

output:

487765455

result:

ok 1 number(s): "487765455"

Test #12:

score: 0
Accepted
time: 582ms
memory: 42864kb

input:

183092 100000000

output:

231786691

result:

ok 1 number(s): "231786691"

Test #13:

score: 0
Accepted
time: 584ms
memory: 41768kb

input:

197291 937201572

output:

337054675

result:

ok 1 number(s): "337054675"

Test #14:

score: 0
Accepted
time: 588ms
memory: 39008kb

input:

200000 328194672

output:

420979346

result:

ok 1 number(s): "420979346"

Test #15:

score: 0
Accepted
time: 593ms
memory: 40956kb

input:

200000 1000000000

output:

961552572

result:

ok 1 number(s): "961552572"

Extra Test:

score: 0
Extra Test Passed