題目描述
在X森林里，上帝創(chuàng)建了生命之樹。

他給每棵樹的每個(gè)節(jié)點(diǎn)（葉子也稱為一個(gè)節(jié)點(diǎn)）上，都標(biāo)了一個(gè)整數(shù)，代表這個(gè)點(diǎn)的和諧值。
上帝要在這棵樹內(nèi)選出一個(gè)非空節(jié)點(diǎn)集S，使得對于S中的任意兩個(gè)點(diǎn)a,b，都存在一個(gè)點(diǎn)列 {a, v1, v2, …, vk, b} 使得這個(gè)點(diǎn)列中的每個(gè)點(diǎn)都是S里面的元素，且序列中相鄰兩個(gè)點(diǎn)間有一條邊相連。

在這個(gè)前提下，上帝要使得S中的點(diǎn)所對應(yīng)的整數(shù)的和盡量大。
這個(gè)最大的和就是上帝給生命之樹的評分。

經(jīng)過atm的努力，他已經(jīng)知道了上帝給每棵樹上每個(gè)節(jié)點(diǎn)上的整數(shù)。但是由于 atm 不擅長計(jì)算，他不知道怎樣有效的求評分。他需要你為他寫一個(gè)程序來計(jì)算一棵樹的分?jǐn)?shù)。
輸入
第一行一個(gè)整數(shù) n 表示這棵樹有 n 個(gè)節(jié)點(diǎn)。
第二行 n 個(gè)整數(shù)，依次表示每個(gè)節(jié)點(diǎn)的評分。
接下來 n-1 行，每行 2 個(gè)整數(shù) u, v，表示存在一條 u 到 v 的邊。由于這是一棵樹，所以是不存在環(huán)的。
輸出
輸出一行一個(gè)數(shù)，表示上帝給這棵樹的分?jǐn)?shù)。
樣例輸入
5
1 -2 -3 4 5
4 2
3 1
1 2
2 5
樣例輸出
8
題目挺不人道的。題干中都說了S為非空集合，但是對于所有點(diǎn)的評分都是負(fù)數(shù)的情況，是可以不選的，也就是結(jié)果為0.這又是為什么呢？
思路：樹形dp的基礎(chǔ)題型，對于以x為頂點(diǎn)的子樹，我們記錄的這這棵子樹中的最大價(jià)值，并且返回給父節(jié)點(diǎn)。如果這個(gè)最大價(jià)值是正數(shù)，就加上，否則就不加。
代碼如下：

#include<bits/stdc++.h> #define ll long long #define inf 1e18 using namespace std;const int maxx=1e5+100; struct edge{int to,next; }e[maxx<<1]; int head[maxx<<1]; ll cor[maxx]; int n,tot; ll ans=0;inline void init() {memset(head,-1,sizeof(head));tot=0; } inline void add(int u,int v) {e[tot].to=v,e[tot].next=head[u],head[u]=tot++; } inline ll dfs(int u,int f) {ll _max=(ll)cor[u];for(int i=head[u];i!=-1;i=e[i].next){int to=e[i].to;if(to==f) continue;ll sum=dfs(to,u);if(sum>0) _max+=sum;}ans=max(ans,_max);return _max; } int main() {scanf("%d",&n);init();for(int i=1;i<=n;i++) scanf("%lld",&cor[i]);int x,y;for(int i=1;i<n;i++){scanf("%d%d",&x,&y);add(x,y);add(y,x);}dfs(1,0);cout<<ans<<endl;return 0; }

努力加油a啊，(^o)/~

總結(jié)

以上是生活随笔為你收集整理的[蓝桥杯][2015年第六届真题]生命之树（树形dp）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。