The queue uvalive 5002 light 1382 uva 12257 （dfs加组合数学）

题目大概是说给一棵树的n个结点从1到n编号，要求每个结点的编号大于其父结点，问有多少种编号方式。

想了挺久的，感觉有点眉目，最后画了下样例YY出解法：

首先求出以每个结点为根的子树大小，记为size[u]，这个DFS一遍就可以求出来；
接下来，dp[u]表示给以u为根的子树size[u]个编号有几种编号方案 ；
然后考虑转移方程：
比如一个结点u有3个儿子v1,v2,v3，那么u子树有size[u]个编号，编号最小的就属于u，剩下size[u]-1分配给u的三个子树，分配方式就有：
C(size[u]-1,size[v1])*C(size[u]-1-size[v1],size[v2])*C(size[u]-1-size[v1]-size[v2],size[v3]) 种

而v1、v2和v3子树对它们被分配的编号又分别有dp[v1]、dp[v2]和dp[v3]种编号方案，因此u子树的size[u]个编号总共的编号方式即dp[u]是：
dp[u] = dp[v1]*dp[v2]*dp[v3]*C(size[u]-1,size[v1])*C(size[u]-1-size[v1],size[v2])*C(size[u]-1-size[v1]-size[v2],size[v3])

于是就是这样用DP求解的。另外C(n,m)可以利用组合数的递推式C(n,m)=C(n-1,m)+C(n-1,m-1)预处理出来。

转载自:http://www.mamicode.com/info-detail-1189403.html

第一种写法 预处理 on^2;

#include <bits/stdc++.h>
using namespace std;
const int mod = 1e9+7;
struct Edge
{
    int v,next;
}edge[1111];
int tot;
typedef long long ll;
ll C[1111][1111];
int head[1111];
int size[1111];
void addedge(int u,int v)
{
    edge[tot].v=v;edge[tot].next=head[u];head[u]=tot++;
}
int getsize(int u)
{
    if(size[u]) return size[u];
    int res=1;
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].v;
        res+=getsize(v);
    }
    return size[u]=res;
}

ll dfs(int u)
{
    int cnt=size[u]-1;
    ll res=1;
    for(int i=head[u];i!=-1;i=edge[i].next)
        {
            int v=edge[i].v;
            res*=(dfs(v)*C[cnt][size[v]])%mod;
            res%=mod;
            cnt-=size[v];
        }
    return res;
}
int vis[10101];
int main()
{
    int t;int n;
    scanf("%d",&t);
    C[0][0]=1;
    for(int i=1;i<=1111;i++)
    {
        C[i][0]=1;
        for(int j=1;j<=1111;j++)
        {
            C[i][j]=(C[i-1][j-1]+C[i-1][j])%mod;
            C[i][j]%=mod;
        }
    }
    for(int cse=1;cse<=t;cse++)
    {
        cin>>n;
        tot=0;
        memset(head,-1,sizeof(head));
        memset(vis,0,sizeof(vis));
        memset(size,0,sizeof(size));
        for(int i=1;i<n;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            addedge(a,b);
            vis[b]=1;
        }
        int root;
        for(int i=1;i<=n;i++)
        {
            if(!vis[i]) {
                root=i; break;
            }
        }
        getsize(root);
        printf("Case %d: %lld\n",cse,dfs(root) );
    }

}

第二种写法 利用乘法逆元 预处理 on

#include <bits/stdc++.h>
using namespace std;
const int mod = 1e9+7;
struct Edge
{
    int v,next;
}edge[1111];
int tot;
typedef long long ll;
int head[1111];
int size[1111];
ll fac[1111];
ll inv[1111];
const int maxn=1111;
ll quickpower(ll a,ll b){
    ll ans=1;
    while(b){
        if(b&1){
            ans*=a;
            ans%=mod;
        }
        a=(a*a)%mod;
        b=b>>1;
    }
    return ans;
}

ll C(int a,int b)
{
    ll res;
    res=fac[a]*inv[b]%mod*inv[a-b]%mod;
    return res;
}
void addedge(int u,int v)
{
    edge[tot].v=v;edge[tot].next=head[u];head[u]=tot++;
}
int getsize(int u)
{
    if(size[u]) return size[u];
    int res=1;
    for(int i=head[u];i!=-1;i=edge[i].next)
    {
        int v=edge[i].v;
        res+=getsize(v);
    }
    return size[u]=res;
}

ll dfs(int u)
{
    int cnt=size[u]-1;
    ll res=1;
    for(int i=head[u];i!=-1;i=edge[i].next)
        {
            int v=edge[i].v;
            res*=(dfs(v)*C(cnt,size[v]))%mod;
            res%=mod;
            cnt-=size[v];
        }
    return res;
}
int vis[10101];

int main()
{
    int t;int n;
    scanf("%d",&t);
    fac[0]=1;
    for(int i=1;i<maxn;i++)
        fac[i]=(fac[i-1]*i)%mod;
     inv[maxn-1]=quickpower(fac[maxn-1],mod-2);
     for(int i=maxn-2;i>=0;i--)
        inv[i]=inv[i+1]*(i+1)%mod;
    for(int cse=1;cse<=t;cse++)
    {
        cin>>n;
        tot=0;
        memset(head,-1,sizeof(head));
        memset(vis,0,sizeof(vis));
        memset(size,0,sizeof(size));
        for(int i=1;i<n;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            addedge(a,b);
            vis[b]=1;
        }
        int root;
        for(int i=1;i<=n;i++)
        {
            if(!vis[i]) {
                root=i; break;
            }
        }
        getsize(root);
        printf("Case %d: %lld\n",cse,dfs(root) );
    }

}