记录编号	199580	评测结果	AAAAAAAAAA
题目名称	火车站饭店	最终得分	100
用户昵称	devil	是否通过	通过
代码语言	C++	运行时间	0.533 s
提交时间	2015-10-26 22:35:09	内存使用	3.36 MiB

#include <cstdlib>
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <map>
#include <stack>
#include <vector>
#include <map>
#include <queue>
#include <ctime>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
typedef unsigned int uint;
const int inf=1061109567;
const int maxn=100010;
const int maxm=70010;
const int mod=1000000007;

vector<int> G[maxn];
int p[maxn],f[maxn],g[maxn];
int vis[maxn],a[maxn];

void build_tree(int root,int fa)
{
    vis[root]=1;p[root]=fa;
    int len=G[root].size();
    for(int i=0;i<len;i++)
    {
        int v=G[root][i];
        if(!vis[v])
        {
            build_tree(v,root);
        }
    }
}

int dp1(int u);    //选 
int dp2(int u);    //不选 

int main()
{
    freopen("profitz.in","r",stdin);
    freopen("profitz.out","w",stdout);
    //clock_t st=clock();
    memset(f,-1,sizeof(f));
    memset(g,-1,sizeof(g));
    int n,u,v;scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    build_tree(1,0);
    int ans=dp1(1);
    ans=max(ans,dp2(1));
    //for(int i=1;i<=n;i++) printf("%d ",f[i]);
    printf("%d\n",ans);
    //clock_t ed=clock();
    //printf("\nTime used : %.5lf Ms\n",double(ed-st)/CLOCKS_PER_SEC);
    return 0;
}

int dp1(int u)
{
    if(f[u]!=-1) return f[u];
    int len=G[u].size();
    int ans=a[u];
    for(int i=0;i<len;i++)
    {
        int v=G[u][i];
        if(p[v]==u)
        {
            ans+=dp2(v);
        }
    }
    return f[u]=ans;
}

int dp2(int u)
{
    if(g[u]!=-1) return g[u];
    int len=G[u].size();
    int ans=0;
    for(int i=0;i<len;i++)
    {
        int v=G[u][i];
        if(p[v]==u)
        {
            ans+=max(dp1(v),dp2(v));
        }
    }
    return g[u]=ans;
}