洛谷P1272 重建道路 树dp

题面

#include<bits/stdc++.h>
using namespace std;
#define N 200
#define reg register
inline void read(int &x){
    
    
	int s=0,w=1;char ch=getchar();
	while(ch<'0'||ch>'9'){
    
    if(ch=='-')w=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){
    
    s=(s<<3)+(s<<1)+(ch&15);ch=getchar();}
	x=s*w;
}
struct node{
    
    
	int to,nxt;
}edge[N<<1];
int n,p,x,y,ans,cnt,siz[N],son[N],head[N],dp[N][N];
inline void addedge(int u, int v){
    
    
	edge[++cnt].to=v,edge[cnt].nxt=head[u],head[u]=cnt;
}
inline void superadd(int u, int v){
    
    
	addedge(u,v),addedge(v,u);
}
void dfs(int u, int fa){
    
    
	siz[u]=1,dp[u][1]=son[u],dp[u][0]=0;
	for(reg int i=head[u];i;i=edge[i].nxt){
    
    
		int vv=edge[i].to;
		if(vv==fa)continue;
		dfs(vv,u);
		siz[u]+=siz[vv];
		for(reg int j=siz[u];~j;j--){
    
    
			for(int k=1;k<j&&k<siz[vv]+1;k++)
				dp[u][j]=min(dp[u][j],dp[u][j-k]+dp[vv][k]-1);
		}
	}
}
int main(){
    
    
	read(n),read(p);
	for(reg int i=1;i<n;i++)read(x),read(y),superadd(x,y),son[x]++;
	memset(dp,0x3f,sizeof(dp));
	dfs(1,0);ans=dp[1][p];
	for(reg int i=1;i<=n;i++)ans=min(ans,dp[i][p]+1);
	printf("%d\n",ans);
}

猜你喜欢

转载自blog.csdn.net/dhdhdhx/article/details/103040088