問題描述
有一棵 n 個節點的樹,樹上每個節點都有乙個正整數權值。如果乙個點被選擇了,那麼在樹上和它相鄰的點都不能被選擇。求選出的點的權值和最大是多少?
輸入格式
第一行包含乙個整數 n 。
接下來的一行包含 n 個正整數,第 i 個正整數代表點 i 的權值。
接下來一共 n-1 行,每行描述樹上的一條邊。
輸出格式
輸出乙個整數,代表選出的點的權值和的最大值。
樣例輸入5
1 2 3 4 5
1 21 3
2 42 5
樣例輸出
12樣例說明
選擇3、4、5號點,權值和為 3+4+5 = 12 。
資料規模與約定
對於20%的資料, n <= 20。
對於50%的資料, n <= 1000。
對於100%的資料, n <= 100000。
權值均為不超過1000的正整數。
樹形dp可以用dfs來實現。
dp[1][i] 表示取當前節點的最大值。
dp[0][i] 表示不去當前節點的最大值。
由於資料是一棵數,所以從任意節點開始dfs都可以。注意標記。
dp[1][u]+=dp[0][v]; //取當前節點,那麼加上不取鄰接點的dp值
dp[0][u]+=max(dp[0][v],dp[1][v]); //不取當前節點,那麼需要加上取或不取鄰接點dp值的最大值。
因為需要取一層,所以需要用+=。
#include#include#include#include#includeusing namespace std;
#define inf 1<<29
int n,a[100000+5],dp[2][100000+5];
vectorg[100000+5];
bool vis[100000+5];
void dfs(int u)
dp[1][u]+=a[u];
}int main()
memset(vis,0,sizeof vis);
dfs(1);
printf("%d\n",max(dp[0][1],dp[1][1]));
return 0;
}
結點選擇(樹形動態規劃)
現在對於樹形動態規劃理解的還不是很透徹,只能先做幾個題目練習一下 本題很重要的兩個方面乙個是建樹,這個需要邊和點的相互配合,乙個點兩條邊的表示都要標記。第二個很重要的方面是動態規劃的變化,本題首先想到的一成一成的樹,要不是子集的結合,要不就是自己本身,上乙個點要不是自己本身,要不是自己和子節點的子節...
動態規劃 樹形動態規劃 結點選擇
題目 有一棵 n 個節點的樹,樹上每個節點都有乙個正整數權值。如果乙個點被選擇了,那麼在樹上和它相鄰的點都不能被選擇。求選出的點的權值和最大是多少?第一行包含乙個整數 n 接下來的一行包含 n 個正整數,第 i 個正整數代表點 i 的權值。接下來一共 n 1 行,每行描述樹上的一條邊。輸出乙個整數,...
藍橋杯 演算法訓練(四)結點選擇(樹形動態規劃)
問題描述 有一棵 n 個節點的樹,樹上每個節點都有乙個正整數權值。如果乙個點被選擇了,那麼在樹上和它相鄰的點都不能被選擇。求選出的點的權值和最大是多少?輸入格式 第一行包含乙個整數 n 接下來的一行包含 n 個正整數,第 i 個正整數代表點 i 的權值。接下來一共 n 1 行,每行描述樹上的一條邊。...