版权声明:转载请注明出处。 https://blog.csdn.net/TungstenC/article/details/83182408
题目描述
从前有棵树。
找出
个点
。
使得
最小。
算法分析
分析可知,选择的相邻两个点的位置也一定相邻,而一条边不一定不会走第二次。
设
为以
为根的子树中选择节点
时已经选择了
条边,其中以
为端点的链的个数为
时的最小贡献。
对着别的代码写的,理解不太深,有空重写一遍。
代码实现
#include <cstdio>
#include <cstring>
#include <algorithm>
#define Min(x,y) std::min(x,y)
const int maxn=3005;
struct edge {int u,v,w;} edges[maxn<<1];
int head[maxn],nxt[maxn<<1],idx=0;
inline void add(int u,int v,int w) {
edges[++idx]=(edge){u,v,w};
nxt[idx]=head[u];head[u]=idx;
}
int f[maxn][maxn][3],tot[maxn];//以i为根的子树中选i,选了j条边,包含端点数为k
void dfs(int x,int fa) {
tot[x]=1;f[x][0][0]=f[x][0][1]=0;
for(int i=head[x];i;i=nxt[i]) {
edge &e=edges[i];if(e.v==fa) continue;
dfs(e.v,x);
for(int j=tot[x]-1;j>=0;--j)//前面的子树中还选了j条边
for(int k=tot[e.v]-1;k>=0;--k)//当前子节点的子树中选边数
for(int a=2;a>=0;--a)//前面的子树中包含的端点数
for(int b=a;b>=0;--b) {//当前子节点的子树中包含的端点数
int tmp=f[x][j][a-b]+f[e.v][k][b]+e.w*(2-(b&1))/*b是0或2走两次,b是1走一次*/;
f[x][j+k+1][a]=Min(f[x][j+k+1][a],tmp);
}
tot[x]+=tot[e.v];
}
}
int main() {
int n,k;scanf("%d%d",&n,&k);
int m=n-1,u,v,w;idx=0;
while(m--) {
scanf("%d%d%d",&u,&v,&w);
add(u,v,w);add(v,u,w);
}
memset(f,0x3f,sizeof(f));dfs(1,1);
int ans=0x3f3f3f3f;
for(int i=1;i<=n;++i) {
ans=Min(ans,Min(f[i][k-1][0],Min(f[i][k-1][1],f[i][k-1][2])));
}
printf("%d\n",ans);
return 0;
}