藍橋杯第六屆 第10題 生命之樹
阿新 • • 發佈:2019-01-30
在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; }