題目描述
小明對數學飽有興趣,並且是個勤奮好學的學生,總是在課後留在教室向老師請教一些問題。一天他早晨騎車去上課,路上見到一個老伯正在修剪花花草草,頓時想到了一個有關修剪花卉的問題。於是當日課後,小明就向老師提出了這個問題:
一株奇怪的花卉,上面共連有N 朵花,共有N-1條枝幹將花兒連在一起,並且未修剪時每朵花都不是孤立的。每朵花都有一個“美麗指數”,該數越大說明這朵花越漂亮,也有“美麗指數”爲負數的,說明這朵花看着都讓人噁心。所謂“修剪”,意爲:去掉其中的一條枝條,這樣一株花就成了兩株,扔掉其中一株。經過一系列“修剪“之後,還剩下最後一株花(也可能是一朵)。老師的任務就是:通過一系列“修剪”(也可以什麼“修剪”都不進行),使剩下的那株(那朵)花卉上所有花朵的“美麗指數”之和最大。
老師想了一會兒,給出了正解。小明見問題被輕易攻破,相當不爽,於是又拿來問你。
輸入輸出格式
輸入格式:
輸入文件maxsum3.in的第一行一個整數N(1 ≤ N ≤ 16000)。表示原始的那株花卉上共N 朵花。
第二行有N 個整數,第I個整數表示第I朵花的美麗指數。
接下來N-1行每行兩個整數a,b,表示存在一條連接第a 朵花和第b朵花的枝條。
輸出格式:
輸出文件maxsum3.out僅包括一個數,表示一系列“修剪”之後所能得到的“美麗指數”之和的最大值。保證絕對值不超過2147483647。
枚舉根節點。
若某一子樹值爲負則不管他。
爲正則先與maxnum比較後再加入根節點。
划水。
#include<cstdio>
#include<algorithm>
using namespace std;
struct bian{
int next,to;
};
int k,n,max1,h[50001],b[50001];
bool f[50001];
bian a[50001];
int doit(int x)
{
int t=h[x];
if (t==0)
return b[x];
int res=0;
f[x]=true;
while (t)
{
if (f[a[t].to]==0)
{
int q=doit(a[t].to);
if (q>0)
{
max1=max(max1,q);
res+=q;
}
}
t=a[t].next;
}
return res+b[x];
}
void read(int x,int y)
{
k++;
a[k].next=h[x];
a[k].to=y;
h[x]=k;
}
int main()
{
scanf("%d",&n);
for (int i=1;i<=n;i++)
scanf("%d",&b[i]);
for (int i=1;i<=n-1;i++)
{
int x,y;
scanf("%d%d",&x,&y);
read(x,y);
read(y,x);
}
max1=-2e9;
for (int i=1;i<=n;i++)
{
max1=max(max1,doit(i));
}
printf("%d",max1);
return 0;
}