題目描述
小明對數學飽有興趣，並且是個勤奮好學的學生，總是在課後留在教室向老師請教一些問題。一天他早晨騎車去上課，路上見到一個老伯正在修剪花花草草，頓時想到了一個有關修剪花卉的問題。於是當日課後，小明就向老師提出了這個問題：
一株奇怪的花卉，上面共連有N 朵花，共有N-1條枝幹將花兒連在一起，並且未修剪時每朵花都不是孤立的。每朵花都有一個“美麗指數”，該數越大說明這朵花越漂亮，也有“美麗指數”爲負數的，說明這朵花看着都讓人噁心。所謂“修剪”，意爲：去掉其中的一條枝條，這樣一株花就成了兩株，扔掉其中一株。經過一系列“修剪“之後，還剩下最後一株花（也可能是一朵）。老師的任務就是：通過一系列“修剪”（也可以什麼“修剪”都不進行），使剩下的那株（那朵）花卉上所有花朵的“美麗指數”之和最大。
老師想了一會兒，給出了正解。小明見問題被輕易攻破，相當不爽，於是又拿來問你。
輸入輸出格式
輸入格式：
輸入文件maxsum3.in的第一行一個整數N(1 ≤ N ≤ 16000)。表示原始的那株花卉上共N 朵花。
第二行有N 個整數，第I個整數表示第I朵花的美麗指數。
接下來N-1行每行兩個整數a,b，表示存在一條連接第a 朵花和第b朵花的枝條。
輸出格式：
輸出文件maxsum3.out僅包括一個數，表示一系列“修剪”之後所能得到的“美麗指數”之和的最大值。保證絕對值不超過2147483647。
輸入輸出樣例
輸入樣例
7
-1 -1 -1 1 1 1 0
1 4
2 5
3 6
4 7
5 7
6 7
說明
【數據規模與約定】
對於60%的數據，有N≤1000；
對於100%的數據，有N≤16000。
樹上dp，先dfs，回溯的時候處理拿或不拿的情況，具體見代碼
代碼如下
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int size = 200100;
int n;
int head[size],next[size],dist[size],l[size];
int tot = 1;
void build(int f,int t)
{
l[tot] = t;
next[tot] = head[f];
head[f] = tot ++;
}
void dfs(int p,int fa)
{
for(int i = head[p] ; i ; i = next[i])
{
int t = l[i];
if(t != fa)
{
dfs(t,p);
dist[p] = max(dist[p],dist[p] + dist[t]);
}
}
}
int main()
{
scanf("%d",&n);
for(int i = 1 ; i <= n ; i ++)
scanf("%d",&dist[i]);
for(int i = 1 ; i < n ; i ++)
{
int f,t;
scanf("%d%d",&f,&t);
build(f,t);
build(t,f);
}
dfs(1,-1);
int ans = 0;
for(int i = 1 ; i <= n ; i ++)
ans = max(ans,dist[i]);
printf("%d\n",ans);
return 0;}