【蓝桥杯 算法训练】强力党逗志芃

题目

题目链接
蓝桥杯 算法训练 强力党逗志芃

思路：

树形dp

定义状态：

dp[i][j]表示以i为根，消耗j个技能点，最多能换取的v

状态转移方程：

dp[pos][k]=max(dp[son][h]+dp[pos][k-h],dp[pos][k])

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int m,n,f[205],v[205];
struct ss{
	int to,nex;
}edge[205];int head[205],ecnt;
void add(int x,int y){
	edge[++ecnt]=(ss){y,head[x]};head[x]=ecnt;
}
//dp[i][j]表示以i为根，消耗j个技能点，最多能换取的v
int dp[205][205];
//dp[pos][k]=max(dp[son][h]+dp[pos][k-h],dp[pos][k])
void dfs(int pos){
	for(int i=head[pos];i;i=edge[i].nex){
		int to=edge[i].to;
		dfs(to);
		for(int k=m;k;k--)
		for(int h=k;h;h--)
			dp[pos][k]=max(dp[to][h]+dp[pos][k-h],dp[pos][k]);
		//这里的dp[pos][k-h]其实代表了循环过的儿子的能力 
		//k倒着循环的原因：保证用当前儿子更新过的都是dp[pos][>k]，以保证dp[pos][k-h]是其他儿子的贡献，不受当前儿子自己影响 
		//h正着循环倒着循环无所谓 
	}
	for(int i=m;i>=1;i--) dp[pos][i]=dp[pos][i-1]+v[pos];//把自己的value加上 
	return;
} 

DFS写dp

int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>v[i];
	for(int i=1;i<=n-1;i++){
		int x,y;cin>>x>>y;add(x,y);f[y]=x;
	}int S;
	for(int i=1;i<=n;i++)
	if(!f[i]) S=i,i=n+1;//技能树的root 
	dfs(S);//树形dp 
	cout<<dp[S][m];return 0;
}