洛谷 P1272重建道路 (树形DP)

23 篇文章 0 订阅

题目大意:
有一棵树,我们去掉任意边,去掉边后剩下的节点的数等于p。问我们最少去掉多少的边。
解题思路:
这种题目是典型的树形DP,每一个节点有点类似于背包。背包的特点就是:到达某个地方选和不选,这里同样的我们也是到达某个节点后选和不选。怎么选?选什么呢?
这题是这样的:dp[u][j]= min (dp[u][j],dp[u][j-k]+dp[v][k])
其中:u代表我们到达了树节点u,j代表我们去掉边后剩下j个节点,dp[u][j]就是最少的删除边数,v是和u邻接的节点。这种树形DP,需要填充DP表的时候需要不断DFS到达叶子结点,这里的dp表边界条件是:
dp[u][1]=deg[u] ,其中deg表示节点的度数。
树形DP的模板是这样的:

void dfs(int u){
	dp[][]=...    //表示边界条件
	for(auto nx){
		dfs(nx);
		for(int k=max;k>=1;k--)	//注意这里必须从大到小!这是树形DP的特点。
			for(int i=1;i<k;i++){
				dp[u][k]= ...//转移
}
}
}

AC代码

#include <bits/stdc++.h>
using namespace std;
const int MAXN=160;
vector<vector<int>> gra(MAXN);
int dp[MAXN][MAXN];
int deg[MAXN];
int n,p;
const int INF=1e9;
int res=INF;
void dfs(int u){
	dp[u][1]=deg[u];
	for(int i=0;i<(int)gra[u].size();i++){
		int nx=gra[u][i];
		dfs(nx);
		for(int j=p;j>=1;j--){
			for(int k=1;k<j;k++){
				dp[u][j]=min(dp[u][j],dp[u][j-k]+dp[nx][k]-2);
			}
		}
	}
	res=min(res,dp[u][p]);
}
int main(){
	cin>>n>>p;
	set<pair<int,int>> ms;
	for(int i=0;i<MAXN;i++)deg[i]=0;
	for(int i=0;i<n-1;i++){
		int a,b;cin>>a>>b;
		if(a>b)swap(a,b);
		if(ms.count({a,b}))continue;
		else ms.insert({a,b});
		a-=1;
		b-=1;
		gra[a].emplace_back(b);
		deg[a]++;
		deg[b]++;
	}
	for(int i=0;i<MAXN;i++)
		for(int j=0;j<MAXN;j++)
			dp[i][j]=INF;
	dfs(0);
	
	cout<<res<<endl;
	return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值