ZOJ 3805 树形dp

题意：有一些机器用一些管子链接 然后每一个机器要有两个入口一个出口 给定了链接的长度是无限的

求最小的宽度

解法：比较二叉树中子树进行存放时的大小，记录节点的最小宽度即可

#include<iostream>
#include<cstdio>
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
#define maxn 111111
struct node{
    int ls,rs;
    int size;
    node(int ls=-1,int rs=-1,int size=0):ls(ls),rs(rs),size(size){}
}e[maxn];
void dfs(int rt){
//    printf("rt:%d %d %d\n",rt,e[rt].ls,e[rt].rs);
    int lsize,rsize;
    if(e[rt].ls==-1){
        lsize=0;
    }else {
        dfs(e[rt].ls);
        lsize=e[e[rt].ls].size;
    }
    if(e[rt].rs==-1){
        rsize=0;
    }else{
        dfs(e[rt].rs);
        rsize=e[e[rt].rs].size;
    }

    e[rt].size=max(max(lsize,rsize),min(lsize,rsize)+1);
//    printf("%d %d %d %d\n",rt,lsize,rsize,e[rt].size);
}
int n,vis[maxn],x;
int main(){
    while(~scanf("%d",&n)){
        memset(vis,0,sizeof vis);
        for(int i=0;i<=n+1;++i){
            e[i].ls=e[i].rs=-1;
            e[i].size=0;
        }
        for(int i=1;i<=n-1;++i){
            scanf("%d",&x);
            if(!vis[x]){
                e[x].ls=i+1;
                vis[x]=1;
            }else{
                e[x].rs=i+1;
            }
        }

        dfs(1);
        printf("%d\n",e[1].size);
    }
    return 0;
}