BZOJ2111 ZJOI2010排列计数

根据Pi>Pi/2可以看出来这是一个二叉树

所以我们可以用树形DP的思想

f[i]=f[i<<1]*f[i<<1|1]*C(s[i]-1,s[i<<1])，s是子树大小

然后求组合数可以用卢卡斯定理

BZ上加强数据后我那个线性求n！逆元就挂掉了，于是就直接算了。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e6+;
ll f[N<<],fac[N],inv[N],s[N<<];
ll n,mod;
ll qmod(ll a,ll b)
{
    ll ans=;
    while(b)
    {
        if(b&)ans=ans*a%mod;
        a=a*a%mod;b>>=;
    }
    return ans;
}
ll C(ll a,ll b)
{
    if(a<b)return ;
    if(a<mod&&b<mod)return fac[a]*qmod(fac[b]*fac[a-b]%mod,mod-)%mod;
    return C(a/mod,b/mod)*C(a%mod,b%mod)%mod;
}
int main()
{
    scanf("%lld%lld",&n,&mod);
    fac[]=;
    for(int i=;i<=n;++i)fac[i]=i*fac[i-]%mod;
    for(int i=n;i;--i)
    {
        s[i]=s[i<<]+s[(i<<)|]+;f[i]=;
        if((i<<)<=n)f[i]=f[i]*f[i<<]%mod;
        if((i<<|)<=n)f[i]=f[i]*f[i<<|]%mod;
        f[i]=f[i]*C(s[i]-,s[i<<])%mod;
    }
    printf("%lld",f[]);
    return ;
}