ZOJ 3805 (树形DP)

题目链接: http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=5337

题目大意：方块连接，呈树形。每个方块有两种接法，一种接在父块边上，宽度+1,一种接在父块上面，宽度+0。且一个母块最多有2个子块。问全局的宽度最小是多少。

解题思路：

对于一个方块，就两种接法。

设dp[i][0]=0,表示接在父块上面。

dp[i][1]=1，表示接在父块边上。

对于一个父块，如果没有子块，宽度不变。

如果有一个子块，肯定接在上面，加上子块的宽度。

如果有两个子块，则有两种解法，max(左1上2),max(左2上1)， 注意这里两个子块不是加起来，因为只要把管子伸长，让两个子块相互重叠掉一部分宽度，所以取max

那么接这两个子块就得加上这两种解法的较小值。（min）

最后结果是dp[1][1]，和dp[1][0]没关系，因为1块肯定要占一个宽度。

 

#include "cstdio"

#include "iostream"

#include "cstring"

using namespace std;

#define maxn 10005

struct Edge

{

    int next,to;

}e[maxn];

int dp[maxn][2],head[maxn],tol;

void addedge(int u,int v)

{

    e[tol].to=v;

    e[tol].next=head[u];

    head[u]=tol++;

}

void dfs(int root)

{

    int cnt=0,l1=0,l2=0,r1=0,r2=0;

    dp[root][0]=0,dp[root][1]=1;

    for(int a=head[root];a!=-1;a=e[a].next) dfs(e[a].to);

    for(int a=head[root];a!=-1;a=e[a].next)

    {

        int t=e[a].to;

        if(!cnt) {l1=dp[t][0],l2=dp[t][1];}

        else {r1=dp[t][0],r2=dp[t][1];}

        cnt++;

    }

    if(!cnt) return;

    else if(cnt==1) {dp[root][0]+=l1;dp[root][1]+=l1;}

    else

    {

        int a=max(l1,r2),b=max(l2,r1);

        dp[root][0]+=min(a,b);dp[root][1]+=min(a,b);

    }

}

int main()

{

    //freopen("in.txt","r",stdin);

    int n,u;

    while(scanf("%d",&n)!=EOF)

    {

        memset(head,-1,sizeof(head));

        tol=0;

        for(int i=2;i<=n;i++)

        {

            scanf("%d",&u);

            addedge(u,i);

        }

        dfs(1);

        printf("%d\n",dp[1][1]);

    }

}

 

3808409

2014-10-20 17:37:23

Accepted

3805

C++

800

468

Physcal