zoukankan      html  css  js  c++  java
  • CF995F Cowmpany Cowmpensation DP+容斥

    题意:给定一颗树,要求每个点的点权范围是 $[1,D]$ 且权值不大于父亲,求方案数.           

    显然,即使 $D$ 再大,$n$ 个点也最多只会取到 $n$ 个值.    

    令 $f[x][i]$ 表示以 $x$ 为根的子树中点 $x$ 取到权值 $i$ 的方案数.   

    令 $s[x][i]$ 表示 $f[x][i]$ 的前缀和.  

    那么有 $f[x][i]=prod s[son][i]$.             

    这里有两种处理方法: 

    1. $f[x][i]$ 是一个关于 $i$ 的最高次项为 $n$ 的 $n+1$ 次多项式,所以算出 $f[1][1]....f[1][n+1]$ 后拉格朗日插值求一下即可.    

    2. 令 $g[i]$ 表示恰好用到 $[1,i]$ 的方案数,那么有 $g[i]=f[i]-sum_{j=1}^{i-1} g[j] imes inom{i-1}{j-1}$.   

    求出 $g[i]$ 后 $Ans=sum_{i=1}^{n} inom{D}{i} g[i]$.   

    #include <cstdio>     
    #include <cstring> 
    #include <string>
    #include <vector>   
    #include <algorithm>    
    #define N 3004  
    #define ll long long 
    #define mod 1000000007 
    #define setIO(s) freopen(s".in","r",stdin) 
    using namespace std;   
    int n,D,g[N],f[N][N],s[N][N],fac[N],inv[N]; 
    vector<int>G[N];     
    int qpow(int x,int y)
    {
        int tmp=1; 
        for(;y;y>>=1,x=(ll)x*x%mod) 
            if(y&1) tmp=(ll)tmp*x%mod;  
        return tmp; 
    } 
    int C(int x,int y) { return (ll)fac[x]*inv[y]%mod*inv[x-y]%mod; }  
    void dfs(int x) 
    {    
        for(int i=1;i<=n;++i) f[x][i]=1;   
        for(int i=0;i<G[x].size();++i) 
        {
            int y=G[x][i];  
            dfs(y);          
            for(int j=1;j<=n;++j) f[x][j]=1ll*f[x][j]*s[y][j]%mod;   
        }  
        for(int i=1;i<=n;++i) s[x][i]=(ll)(s[x][i-1]+f[x][i])%mod;  
    }
    int main() 
    { 
        // setIO("input");  
        int i,j; 
        scanf("%d%d",&n,&D);  
        for(i=2;i<=n;++i) 
        {
            int ff; 
            scanf("%d",&ff),G[ff].push_back(i);   
        }    
        dfs(1);              
        fac[0]=inv[0]=1;  
        for(i=1;i<=n;++i) fac[i]=(ll)i*fac[i-1]%mod, inv[i]=qpow(fac[i],mod-2); 
        for(i=1;i<=n;++i) 
        {
            g[i]=f[1][i];   
            for(j=1;j<i;++j)             
                g[i]=(ll)(g[i]-(ll)C(i-1,i-j)*g[j]%mod+mod)%mod; 
        }
        int ans=0;   
        int tmp=1;  
        for(i=1;i<=min(D,n);++i) 
        { 
            tmp=(ll)tmp*qpow(i,mod-2)%mod*(D-i+1)%mod;   
            ans=(ll)(ans+(ll)tmp*g[i]%mod)%mod;   
        }   
        printf("%d
    ",ans); 
        return 0;
    }
    

      

  • 相关阅读:
    P4342 [IOI1998]Polygon
    P1194 买礼物
    P1363 幻想迷宫
    Installing Wine 1.5: configure: error: Cannot build a 32-bit program, you need to install 32-bit development libraries(转载)
    Linux系统调用之open(), close() (转载)
    undefined reference to 'pthread_create'问题解决(转载)
    linux中的C里面使用pthread_mutex_t锁(转载)
    #if、#ifdef、#if defined之间的区别(转载)
    linux下解压tgz文件(转载)
    linux修改用户主目录的方法 (转载)
  • 原文地址:https://www.cnblogs.com/guangheli/p/12174561.html
Copyright © 2011-2022 走看看