hdu1520（poj2342）（树形dp）

这道题的题目和poj2342是一样的，然而数据量比那道题大，导致我那道题直接交到这个上是TLE，然后呢，对于这道题目，应该用链式前向星来写，这里我用的是双向建边，因为只要是有直接关系就不行，那么我可以认为不用管谁是上级，所以可以双向建边，注意当双向建边的时候数据量是原变数的2倍，否则会RE

#include <iostream>
#include <string.h>
#include <stdio.h>
//#pragma comment(linker, "/STACK:10240000,10240000")//扩栈
const int maxn=12005;
int n,edge;
bool vis[maxn];
int dp[maxn][2],pre[maxn],head[maxn];
using namespace std;
struct node
{
    int to;
    int next;
};
node edges[maxn];
void init()
{
    memset(vis,0,sizeof(vis));
    memset(dp,0,sizeof(dp));
    memset(pre,0,sizeof(pre));
    memset(head,-1,sizeof(head));
    memset(edges, -1, sizeof(edges));
    edge=0;
}
void addedge(int u,int v)
{
    edges[edge].to=v;
    edges[edge].next=head[u];
    head[u]=edge++;
}
void tree_dp(int x)
{
    vis[x]=1;
    for(int k=head[x]; k!=-1; k=edges[k].next)
    {
        int i=edges[k].to;
        if(!vis[i])
        {
            tree_dp(i);
            dp[x][1]+=dp[i][0];//boss来了，下属没来
            dp[x][0]+=max(dp[i][0],dp[i][1]);//boss没来，下属可能来也可能没来
        }
    }
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        init();
        int a,b,q=0,t=0;
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&dp[i][1]);
            //pre[i]=i;//poj的数据这里有bug，要不然我早就过了，可能会出现不是1到n的数，比如4 1,1,1,1 2，7 5， 2就没法解决，但是我个人认为这个地方应该没问题的，是数据的问题
        }
        while(scanf("%d%d",&a,&b)!=EOF)
        {
            if(a==0||b==0)
                break;
            addedge(a,b);
            addedge(b,a);
            pre[a]=b;
            q=b;
        }
        tree_dp(q);
        printf("%d\n",max(dp[q][0],dp[q][1]));
    }
    return 0;
}