QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#416601 | #8715. 放苹果 | forget-star | AC ✓ | 593ms | 42864kb | C++14 | 3.4kb | 2024-05-21 23:52:51 | 2024-05-21 23:52:51 |
Judging History
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,我给组数据试试?
詳細信息
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