HDU 1520 Anniversary party 树形DP

每个节点有权值，子节点和父节点不能同时选，问最后能选的最大价值是多少？ 输入格式有点傻逼。

//#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<iostream>
#include<sstream>
#include<cmath>
#include<climits>
#include<string>
#include<map>
#include<queue>
#include<vector>
#include<stack>
#include<set>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
#define pb(a) push_back(a)
#define INF 0x1f1f1f1f
#define lson idx<<1,l,mid
#define rson idx<<1|1,mid+1,r
#define PI  3.1415926535898
template<class T> T min(const T& a,const T& b,const T& c) {
    return min(min(a,b),min(a,c));
}
template<class T> T max(const T& a,const T& b,const T& c) {
    return max(max(a,b),max(a,c));
}
void debug() {
#ifdef ONLINE_JUDGE
#else

    freopen("d:\\in.txt","r",stdin);
   // freopen("d:\\out1.txt","w",stdout);
#endif
}
int getch() {
    int ch;
    while((ch=getchar())!=EOF) {
        if(ch!=' '&&ch!='\n')return ch;
    }
    return EOF;
}

vector<int> g[6666];
int val[6666];
int dp[6666][2];
int vis[6666][2];
int f(int u,int st,int fa)
{
    if(vis[u][st])return dp[u][st];
    vis[u][st]=1;

    int &res=dp[u][st];
    res=0;

    int num=0;
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        if(v!=fa)
            num+=f(v,0,u);
    }
    res=max(res,num);


    if(st==0||fa==-1)
    {
        num=val[u];
        for(int i=0;i<g[u].size();i++)
        {
            int v=g[u][i];
            if(v!=fa)
                num+=f(v,1,u);
        }
        res=max(num,res);
    }
    return res;
}
void read(int n)
{
    for(int i=1;i<=n;i++)
        g[i].clear();
    for(int i=1;i<=n;i++)
        scanf("%d",&val[i]);
    for(int i=1;i<=n;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        g[a].push_back(b);
        g[b].push_back(a);
    }
}


int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        read(n);
        memset(vis,0,sizeof(vis));
        int num=f(1,0,-1);
        printf("%d\n",num);
    }
    return 0;
}