演算法訓練 結點選擇 樹形動態規劃

2021-08-16 02:34:13 字數 1162 閱讀 7629

問題描述

有一棵 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 行,每行描述樹上的一條邊。...