题目大意:给你一颗树,树上每个点有一个点权,让你从1号点走k步,问可以得到的最大点权和是多少?
思路:这道题既然是个树结构,我们很容易想到是树形dp
设计状态:
由于我们走到一个点有可能是要回到这个点走到另一颗子树上去,也有可能不回来。所以要设两个状态:
1.
d
p
[
0
]
[
x
]
[
s
]
dp[0][x][s]
dp[0][x][s]表示以x为根节点的子树走s步回到x点可以得到的最大点权和
2.
d
p
[
1
]
[
x
]
[
s
]
dp[1][x][s]
dp[1][x][s]表示以x为根节点的子树走s步不回到x点可以得到的最大点权和
设计转移:
这里由于我们并不关心我们这个最大值是怎么走出来的,那么我们就可以用背包的思想,对于一颗子树,以根节点的儿子为根的子树所获得的最大点权和作为价值,枚举出所有给它的步数(步数作为体积)来进行转移,并且用已经转移的所有子树看作一个整体与新考虑的子树合并
dp转移方程:k为x的一个儿子
1.走到x点,回到x点:
d
p
[
0
]
[
x
]
[
j
+
2
]
=
m
a
x
(
d
p
[
0
]
[
x
]
[
j
+
2
]
,
d
p
[
0
]
[
k
]
[
t
]
+
d
p
[
0
]
[
x
]
[
j
−
t
]
dp[0][x][j+2]=max(dp[0][x][j+2],dp[0][k][t]+dp[0][x][j-t]
dp[0][x][j+2]=max(dp[0][x][j+2],dp[0][k][t]+dp[0][x][j−t]
这里之所以要+2,是因为从根节点到儿子再回来要用两步
2.不回到x点,但是回到k点之后要回到x点以向剩下的子树走:
d
p
[
1
]
[
x
]
[
j
+
2
]
=
m
a
x
(
d
p
[
1
]
[
x
]
[
j
+
2
]
,
d
p
[
1
]
[
x
]
j
−
t
]
+
d
p
[
0
]
[
k
]
[
t
]
dp[1][x][j+2]=max(dp[1][x][j+2],dp[1][x]j-t]+dp[0][k][t]
dp[1][x][j+2]=max(dp[1][x][j+2],dp[1][x]j−t]+dp[0][k][t]
3.不回到x点,也不会到k点,一直走下去:
d
p
[
1
]
[
x
]
[
j
+
1
]
=
m
a
x
(
d
p
[
1
]
[
x
]
[
j
+
1
]
,
d
p
[
0
]
[
x
]
[
j
−
t
]
+
d
p
[
1
]
[
k
]
[
t
]
dp[1][x][j+1]=max(dp[1][x][j+1],dp[0][x][j-t]+dp[1][k][t]
dp[1][x][j+1]=max(dp[1][x][j+1],dp[0][x][j−t]+dp[1][k][t]
我们是遍历完别的要回到根节点的(也就是2情况)之后再遍历这个子树,所以要加的是
d
p
[
0
]
[
x
]
[
j
−
t
]
dp[0][x][j-t]
dp[0][x][j−t]而不是
d
p
[
1
]
[
x
]
[
j
−
t
]
dp[1][x][j-t]
dp[1][x][j−t]
下附AC代码:
#include<iostream>
#include<vector>
#include<cstdio>
#include<cstring>
using namespace std;
int read(){
char s;
int x=0,f=1;
s=getchar();
while(s<'0'||s>'9'){
if(s=='-')f=-1;
s=getchar();
}
while(s>='0'&&s<='9'){
x*=10;
x+=s-'0';
s=getchar();
}
return x*f;
}
int n,m;
int c[110];
vector<int>v[110];
int dp[2][110][210];//01分别代表是否回到该节点,并以该节点为根的子树用k步可以得到的最大价值
void dfs(int x,int fa){
for(int i=0;i<=m;i++){
dp[0][x][i]=dp[1][x][i]=c[x];
}
for(int i=0;i<v[x].size();i++){
int k=v[x][i];
if(k==fa)continue;
dfs(k,x);
for(int j=m;j>=0;j--){
for(int t=0;t<=j;t++){
dp[0][x][j+2]=max(dp[0][x][j+2],//从子树回来也回到根节点
dp[0][x][j-t]+dp[0][k][t]);
dp[1][x][j+2]=max(dp[1][x][j+2],//从子树回来,不回到根节点
dp[0][k][t]+dp[1][x][j-t]);
dp[1][x][j+1]=max(dp[1][x][j+1],//不从子树回来,一路走到底
dp[1][k][t]+dp[0][x][j-t]);
}
}
}
}
int main(){
while(cin>>n){
m=read();
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++){
c[i]=read();
v[i].clear();
}
for(int i=1;i<n;i++){
int a,b;
a=read();
b=read();
v[a].push_back(b);
v[b].push_back(a);
}
dfs(1,0);
cout<<max(dp[0][1][m],dp[1][1][m])<<endl;
}
}