在x森林裡,上帝建立了生命之樹。
他給每棵樹的每個節點(葉子也稱為乙個節點)上,都標了乙個整數,代表這個點的和諧值。
上帝要在這棵樹內選出乙個非空節點集s,使得對於s中的任意兩個點a,b,都存在乙個點列 使得這個點列中的每個點都是s裡面的元素,且序列中相鄰兩個點間有一條邊相連。
在這個前提下,上帝要使得s中的點所對應的整數的和盡量大。
這個最大的和就是上帝給生命之樹的評分。
經過atm的努力,他已經知道了上帝給每棵樹上每個節點上的整數。但是由於 atm 不擅長計算,他不知道怎樣有效的求評分。他需要你為他寫乙個程式來計算一棵樹的分數。
「輸入格式」
第一行乙個整數 n 表示這棵樹有 n 個節點。
第二行 n 個整數,依次表示每個節點的評分。
接下來 n-1 行,每行 2 個整數 u, v,表示存在一條 u 到 v 的邊。由於這是一棵樹,所以是不存在環的。
「輸出格式」
輸出一行乙個數,表示上帝給這棵樹的分數。
「樣例輸入」
51 -2 -3 4 5
4 23 1
1 22 5
「樣例輸出」
8「資料範圍」
對於 30% 的資料,n <= 10
對於 100% 的資料,0 < n <= 10^5, 每個節點的評分的絕對值不超過 10^6 。
題目分析:題意一開始沒讀懂,去問的zx學長,簡單來說就是從樹中求乙個最大連通子塊,保證這個連通子塊的權值和最大,因為說子樹不太合適,就叫做連通子塊啦
題目給到了1e5的資料,所以暴力是肯定不能考慮的了,因為涉及到權值和的問題,所以我們不難聯想到字首和,在樹上維護字首和不就是樹形dp嘛,但如果是單純的維護字首和的話,又不太好更新答案,所以我們考慮一下如何設計dp
想了半天還是沒有思路啊,但zx學長看了一眼就給秒掉了,再次不得不承認人與人之間的差距。。
這個題目給出的是一棵無根樹,我們依然可以當做有根樹來做,隨意找乙個節點當做根,跑一遍dfs,dp[i]代表節點i為上邊界的子連通塊的最大值,因為是子連通塊而不是子樹,所以在任意節點我們是可以直接斷開的,為什麼要斷開呢?因為假如節點i的某棵子樹做出的貢獻是負數,那麼我們完全可以捨棄掉這棵子樹,從而使得每次轉移的答案最優,這樣轉移方程也就很容易了:
維護完dp後o(n)跑一遍最大值就是答案啦
這裡需要注意一下,一開始我理解的dp[i]的含義有誤,所以導致整個思路混亂,從而沒有正確的設計出轉移方程,我一開始以為dp[i]代表的是到節點i為止,子連通塊中的最大值,從而出現了另乙個疑問:如果最大值是在沿著節點i的父節點那條鏈上該怎麼辦呢?其實並不用擔心這個問題,如果包含最大值的那一坨真的出現在節點i父節點的那條鏈上,雖然我們不能用節點i來更新最大值,但總有節點是可以更新到這個最大值的,我們可以這樣理解,節點i作為最大值這一坨的下界,自然無法更新答案,但只要是這個最大值合法,那麼他必定存在乙個上界,也是屬於這棵樹中的任意乙個節點,當遍歷到上界的這個節點的時候,就可以更新相應的最大值了,所以在這裡個人認為dp[i]理解為以節點i為上界的最大子連通塊比較合適
**:
#include#include#include#include#include#include#include#include#include#include#include#include#include#include#include#include#includeusing namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int n=1e5+100;
vectornode[n];
ll dp[n];
int val[n];
void dfs(int u,int f)}
int main()
dfs(1,-1);
ll ans=-inf;
for(int i=1;i<=n;i++)
ans=max(ans,dp[i]);
printf("%d\n",ans);
return 0;
}
藍橋杯 生命之樹 樹形DP
生命之樹 在x森林裡,上帝建立了生命之樹。他給每棵樹的每個節點 葉子也稱為乙個節點 上,都標了乙個整數,代表這個點的和諧值。上帝要在這棵樹內選出乙個非空節點集s,使得對於s中的任意兩個點a,b,都存在乙個點列 使得這個點列中的每個點都是s裡面的元素,且序列中相鄰兩個點間有一條邊相連。在這個前提下,上...
生命之樹 樹形DP 藍橋
原題 在x森林裡,上帝建立了生命之樹。他給每棵樹的每個節點 葉子也稱為乙個節點 上,都標了乙個整數,代表這個點的和諧值。上帝要在這棵樹內選出乙個非空節點集 s,使得對於 s 中的任意兩個點 a,b都存在乙個點列 使得這個點列中的每個點都是 s 裡面的元素,且序列中相鄰兩個點間有一條邊相連。在這個前提...
藍橋杯 生命之樹 樹狀dp
在x森林裡,上帝建立了生命之樹。他給每棵樹的每個節點 葉子也稱為乙個節點 上,都標了乙個整數,代表這個點的和諧值。上帝要在這棵樹內選出乙個非空節點集s,使得對於s中的任意兩個點a,b,都存在乙個點列 使得這個點列中的每個點都是s裡面的元素,且序列中相鄰兩個點間有一條邊相連。在這個前提下,上帝要使得s...