1. 程式人生 > >藍橋杯第六屆 第10題 生命之樹

藍橋杯第六屆 第10題 生命之樹

在X森林裡,上帝建立了生命之樹。

他給每棵樹的每個節點(葉子也稱為一個節點)上,都標了一個整數,代表這個點的和諧值。
上帝要在這棵樹內選出一個非空節點集S,使得對於S中的任意兩個點a,b,都存在一個點列 {a, v1, v2, ..., vk, b} 使得這個點列中的每個點都是S裡面的元素,且序列中相鄰兩個點間有一條邊相連。
在這個前提下,上帝要使得S中的點所對應的整數的和儘量大。
這個最大的和就是上帝給生命之樹的評分。

經過的努力,他已經知道了上帝給每棵樹上每個節點上的整數。但是由於不擅長計算,他不知道怎樣有效的求評分。他需要你為他寫一個程式來計算一棵樹的分數。

input
5
1 -2 -3 4 5
4 2
3 1
1 2
2 5

output


8

「資料範圍」
對於 30% 的資料,n <= 10
對於 100% 的資料,0 < n <= 10^5, 每個節點的評分的絕對值不超過 10^6 。


#include<cstdio>
#include<cstring>
#include<algorithm>
#include <iostream>
#include<vector>
using namespace std;
typedef long long LL;
const int maxn=100005;
vector<int> tree[maxn];   //通過容器實現樹的連通
int n;
LL dp[maxn];
int vis[maxn];
LL MAX=0;
void dfs(int u)
{
//	printf("dfs (%d )	",u);
    vis[u]=1;
    LL sum=0;//先把每一個結點的子樹和 初始化為 0
    for(int i=0;i<tree[u].size();i++)
    {
        int v=tree[u][i];
        if(!vis[v])
        {
            dfs(v);         
            if(dp[v]>=0)//把改結點 下面的葉子中,正數求和(不包括自身)
                sum=sum+dp[v];
        }
    }
    if(sum>0)dp[u]+=sum;    //如果該結點 的子樹和大於0 ,則可以加上(因為要想聯通,那摩結點位置必須加上,無論該結點是正數還是負數)
//	printf("dp[%d] = %d\n",u,dp[u]);
    MAX=max(dp[u],MAX);
//	cout<<maxn<<endl;
}
int main()
{
    maxn=-0x3fffffff;
    scanf("%d",&n);
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=n;i++)
        cin>>dp[i];
    for(int i=1;i<=n-1;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        tree[u].push_back(v);   //一條邊,雙向關聯
        tree[v].push_back(u);
    }
    dfs(1);
    cout<<MAX<<endl;
    return 0;
}