有一棵點數為 n 的樹,樹邊有邊權。給你乙個在 0∼n 之內的正整數k,你要在這棵樹中選擇 k 個點,將其染成黑色,並將其他的 n−k個點染成白色。將所有點染色後,你會獲得黑點兩兩之間的距離加上白點兩兩之間距離的和的收益。
問收益最大值是多少。
輸入第一行兩個整數 n,k。
接下來 n−1 行每行三個正整數 fr,to,dis,表示該樹中存在一條長度為 dis 的邊 (fr,to)
輸入保證所有點之間是聯通的。
輸出輸出乙個正整數,表示收益的最大值。
樣例輸入 [複製]
5 21 2 3
1 5 1
2 3 1
2 4 2
樣例輸出 [複製]
17思路
樹形dp
size[i]:i子樹的節點個數
f[i][j]:在i子樹中染j個黑點的最大貢獻
更新時考慮每條邊對答案的貢獻
即:這條邊兩側的黑點個數乘積邊權+兩側白點個數乘積邊權
#include
using
namespace std;
#define int long long
inline
intread()
const
int n=
2010
;int n,k;
struct nodee[n<<1]
;int fir[n]
,cnt=
0,dp[n]
[n],siz[n]
;inline
void
add(
int u,
int v,
int w)
;fir[u]
=cnt;
}inline
intcalc
(int w,
int siz,
int y)
inline
void
dfs(
int u,
int fa)
} siz[u]
+=siz[v];}
}signed
main()
dfs(1,
0);int ans=dp[1]
[k];
printf
("%lld"
,ans)
;return0;
}
HAOI2015 樹上染色
考慮子樹當中所有邊的貢獻即可。然後就能簡單做樹上揹包了。但是要注意列舉的順序,應該從大到小更新,否則某個狀態會多次被加。如果不想考慮列舉順序,那麼直接dp的時候用乙個臨時陣列記錄。ac pragma gcc optimize ofast funroll all loops include defin...
HAOI2015 樹上染色
嘟嘟嘟 首先這一眼看出來,要樹形dp。然後發現狀態不好設,剛開始我想的是dp i j 表示以 i 為根的子樹,選了 j 個黑點的最大價值。結果就不會轉移了。轉移的時候想考慮 這一條邊的貢獻,但是發現這個狀態的轉移所涉及的不只是這一條邊,還有子樹中的邊,於是就徹底gg了。還是看了題解。題解也是考慮貢獻...
HAOI2015 樹上染色
有一棵點數為 n 的樹,樹邊有邊權。給你乙個在 0 n 之內的正整數 k 你要在這棵樹中選擇 k個點,將其染成黑色,並將其他 的n k個點染成白色 將所有點染色後,你會獲得黑點兩兩之間的距離加上白點兩兩之間的距離的和的受益。問受益最大值是多少。輸入格式 第一行包含兩個整數 n,k 接下來 n 1 行...