码迷,mamicode.com
首页 > 其他好文 > 详细

[CTS2019]随机立方体(容斥+组合数学)

时间:2019-05-20 23:03:56      阅读:245      评论:0      收藏:0      [点我收藏+]

标签:注意   printf   bsp   a*   坐标   一个   mes   组合数学   ret   

这题七次方做法显然,但由于我太菜了,想了一会发现也就只会这么多,而且别的毫无头绪。发现直接做不行,那么,容斥!

f[i]为至少i个极值的方案,然后这里需要一些辅助变量,a[i]表示选出i个三维坐标均不相同的i个极大值的方案数,g[i]表示i个极大的数任意一个至少有一维坐标相同的点的个数,h[i]表示g[i]的极值可以同时存在的方案数,那么有f[i]=C(nml,g[i])a[i]h[i](nml-g[i])!。

a[i]很容易求得,就是(∏(n-j)(m-j)(l-j))/i!,其中j∈[0,i),g[i]更好求,就是nml-(n-i)(m-i)(l-i)

然后要进行一些关于上升幂的运算,我这里打不出式子(因为太菜了不会LaTeX),所以就不打了。注意维护g[i]的前缀积,具体细节看code吧。

技术图片
#include<bits/stdc++.h>
using namespace std;
const int N=5e6+7,mod=998244353;
int n,m,l,k,mn,ans,fac[N],inv[N],a[N],f[N],g[N],h[N],pre[N];
int calc(int x){return 1ll*(n-x)*(m-x)%mod*(l-x)%mod;}
int qpow(int a,int b)
{
    int ret=1;
    while(b)
    {
        if(b&1)ret=1ll*ret*a%mod;
        a=1ll*a*a%mod,b>>=1;
    }
    return ret;
}
int C(int a,int b){return 1ll*fac[a]*inv[b]%mod*inv[a-b]%mod;}
int main()
{
    inv[0]=inv[1]=fac[0]=1;for(int i=2;i<N;i++)inv[i]=1ll*(mod-mod/i)*inv[mod%i]%mod;
    for(int i=1;i<N;i++)fac[i]=1ll*fac[i-1]*i%mod,inv[i]=1ll*inv[i-1]*inv[i]%mod;
    int T;scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d%d%d",&n,&m,&l,&k);
        mn=min(n,min(m,l));
        a[0]=h[0]=pre[0]=1;
        for(int i=1;i<=mn;i++)a[i]=1ll*a[i-1]*calc(i-1)%mod;
        for(int i=1;i<=mn;i++)g[i]=(1ll*g[i-1]+calc(i-1)-calc(i)+mod)%mod;
        for(int i=1;i<=mn;i++)pre[i]=1ll*pre[i-1]*g[i]%mod;
        pre[mn]=qpow(pre[mn],mod-2);
        for(int i=mn;i;i--)pre[i-1]=1ll*pre[i]*g[i]%mod;
        for(int i=1;i<=mn;i++)f[i]=1ll*a[i]*pre[i]%mod;
        ans=0;
        for(int i=k;i<=mn;i++)
        if((i-k)&1)ans=(ans-1ll*C(i,k)*f[i]%mod+mod)%mod;
        else ans=(ans+1ll*C(i,k)*f[i])%mod;
        printf("%d\n",ans);
    }
}
View Code

 

[CTS2019]随机立方体(容斥+组合数学)

标签:注意   printf   bsp   a*   坐标   一个   mes   组合数学   ret   

原文地址:https://www.cnblogs.com/hfctf0210/p/10896868.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!