Codeforces Round #384 (Div. 2) D. Chloe and pleasant prizes(树形DP)

题目链接：D. Chloe and pleasant prizes

题意：

给你一棵树，树上每一个点有一个权值，现在让你选两个点a,b，使得a不是b的子节点且b不是a的子节点，问最大的ans。

ans的定义是sum(a)+sum(b)(sum(a)以a为根的所以子节点权值和(包括a)，b同理)。

题解：

dp[i]记录以i这个节点为根的最大子节点的sum，f[i]记录sum[i]，然后dfs下去，中途更新答案即可。

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=a;i<=b;++i)
using namespace std;
typedef long long ll;

const int N=2e5+7;
int a[N],n,g[N],nxt[N*2],v[N*2],ed;
ll dp[N],f[N],ans,inf=1ll<<61;

void adg(int x,int y){v[++ed]=y,nxt[ed]=g[x],g[x]=ed;}
inline void up(ll &a,ll b){if(a<b)a=b;}

void dfs(int u,int fa)
{
    f[u]=a[u],dp[u]=-inf;
    for(int i=g[u];i;i=nxt[i])if(v[i]!=fa)
    {
        dfs(v[i],u);
        if(dp[u]>-inf)up(ans,dp[u]+dp[v[i]]);
        up(dp[u],dp[v[i]]),f[u]+=f[v[i]];
    }
    up(dp[u],f[u]);
}

int main()
{
    scanf("%d",&n);
    F(i,1,n)scanf("%d",a+i);
    F(i,1,n-1)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        adg(x,y),adg(y,x);
    }
    ans=-inf,dfs(1,0);
    if(ans>-inf)printf("%lld\n",ans);
    else puts("Impossible");
    return 0; 
}