hdu5279 YJC plays Minecraft

题目描述

题解：

岛屿之间的边砍/不砍情况有$2^n$种，

但是需要剪掉所有的岛上都首尾相连的情况。

$dp$一下对于完全图没有限制（$f$）/有限制（$g$）的情况数。

方程：$$f[i]=\sum(C(i-1,j-1)*j^{(j-2)}*f[i-j])$$

$$g[i]=\sum(C(i-2,j-2)*j^{(j-2)}*f[i-j])$$

其中$j^(j-2)$是$j$个点的完全图的生成树个数。

打开组合数之后$CDQ$即可。

统计答案时$ans=2^n*\prod(f)-\prod(g)$。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MOD = ;
const int N = ;
typedef long long ll;
template<typename T>
inline void read(T&x)
{
    T f = ,c = ;char ch = getchar();
    while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
    while(ch>=''&&ch<=''){c=c*+ch-'';ch=getchar();}
    x = f * c;
}
ll fastpow(ll x,int y)
{
    ll ret = ;
    while(y)
    {
        if(y&)ret= ret*x%MOD;
        x= x*x % MOD;
        y>>=;
    }
    return ret;
}
int to[*N],lim=,L;
ll W[*N],inv;
void ntt(ll *a,int len,int k)
{
    for(int i=;i<len;i++)
        if(i<to[i])swap(a[i],a[to[i]]);
    for(int i=;i<len;i<<=)
    {
        ll w0 = W[i];
        for(int j=;j<len;j+=(i<<))
        {
            ll w = ;
            for(int o=;o<i;o++,w=w*w0%MOD)
            {
                ll w1 = a[j+o],w2 = a[j+o+i]*w%MOD;
                a[j+o] = (w1+w2)%MOD;
                a[j+o+i] = (w1-w2+MOD)%MOD;
            }
        }
    }
    if(k==-)
    {
        for(int i=;i<(len>>);i++)swap(a[i],a[len-i]);
        for(int i=;i<len;i++)a[i]=a[i]*inv%MOD;
    }
}
ll a[*N],b[*N],c[*N],d[*N],e[*N];
ll f[N],g[N],fj[N],gj[N],jc[N],jn[N];
void clear()
{
    for(int i=;i<lim;i++)a[i]=b[i]=;
}
void work()
{
    ntt(a,lim,),ntt(b,lim,),ntt(c,lim,);
    for(int i=;i<lim;i++)d[i]=a[i]*b[i]%MOD,e[i]=a[i]*c[i]%MOD;
    ntt(d,lim,-),ntt(e,lim,-);
}
void cdq(int l,int r)
{
    if(l==r)
    {
        if(l==)f[l] = ;
        else f[l] = f[l]*jc[l-]%MOD;
        if(l<=)g[l] = ;
        else g[l] = g[l]*jc[l-]%MOD;
        return ;
    }
    int mid = (l+r)>>;
    cdq(l,mid);
    lim = ,L = ;
    while(lim<=(r-l+))lim<<=,L++;
    for(int i=;i<lim;i++)to[i]=((to[i>>]>>)|((i&)<<(L-)));
    inv = fastpow(lim,MOD-);
    for(int i=;i<lim;i<<=)W[i]=fastpow(,(MOD-)/(i<<));
    clear();
    for(int i=;i<=mid-l;i++)a[i]=f[i+l]*jn[i+l]%MOD;
    for(int i=;i<=r-l;i++)b[i]=fj[i],c[i]=gj[i];
    work();
    for(int i=mid-l+;i<=r-l;i++)f[i+l]=(f[i+l]+d[i])%MOD;
    for(int i=mid-l+;i<=r-l;i++)g[i+l]=(g[i+l]+e[i])%MOD;
    cdq(mid+,r);
}
int T,n,x;
int main()
{
    jc[]=jn[]=;
    fj[]=jc[]=jn[]=;
    for(int i=;i<=;i++)
    {
        jc[i] = jc[i-] * i % MOD;
        jn[i] = fastpow(jc[i],MOD-);
        ll tmp = fastpow(i,i-);
        fj[i] = tmp*jn[i-]%MOD;
        gj[i] = tmp*jn[i-]%MOD;
    }
    cdq(,);
    read(T);
    while(T--)
    {
        read(n);
        ll s1=,s2=;
        for(int i=;i<=n;i++)
        {
            read(x);
            s1 = s1 * f[x]%MOD;
            s2 = s2 * g[x]%MOD;
        }
        s1 = s1*fastpow(,n)%MOD;
        printf("%lld\n",(s1-s2+MOD)%MOD);
    }
    return ;
}