题目描述
给定一棵有向树T,树T 中每个顶点u都有一个权w(u);树的每条边(u,v)也都有一个非负边长d(u,v)。有向树T的每个顶点u 可以看作客户,其服务需求量为w(u)。每条边(u,v)的边长d(u,v) 可以看作运输费用。如果在顶点u 处未设置服务机构,则将顶点u 处的服务需求沿有向树的边(u,v)转移到顶点v 处服务机构需付出的服务转移费用为w(u)*d(u,v)。树根处已设置了服务机构,现在要在树T中增设k处服务机构,使得整棵树T 的服务转移费用最小。
编程任务:对于给定的有向树T,编程计算在树T中增设k处服务机构的最小服务转移费用。
输入
第1行有2个正整数n和k。n表示有向树T 的边数;
k是要增设的服务机构数。有向树T 的顶点编号为0,1,…,n。根结点编号为0。
接下来的n行中,每行有表示有向树T的一条有向边的3个整数。第i+1行的3 个整数wi,vi,di分别表示编号为i 的顶点的权为wi,相应的有向边为(i, vi),其边长为di。
输出
最小服务转移费。
样例输入 Copy
4 2
1 0 1
1 1 10
10 2 5
1 2 3
样例输出 Copy
4
实在查不出哪里错了 输出是0
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <queue>
#define N 250
#define K 105
using namespace std;
int n,k,dp[N*2][N][K];//等价二叉树 因为有边权 所以不能按左儿子右兄弟直接转二叉树
struct tree{
int l,r,w,j,f,d[N],deep;
}q[N*2];
int read(){
char c; int x=0,f=1;
c=getchar();
while(c<'0'||c>'9'){
if(c=='-') f=-1;
c=getchar();
}
while(c>='0'&&c<='9'){
x=(x<<1)+(x<<3)+(c-'0');//x*10+c-'0'
c=getchar();
}
return x*f;
}
void loading(){
int fa;
n=read(); k=read();
int num=n;
for(int i=0;i<=2*n;i++) q[i].l=q[i].r=-1;
for(int i=1;i<=n;i++){
q[i].w=read(); fa=read(); q[i].j=read();//顶点的权 父亲节点 边长
if(q[fa].l==-1){//无左子树
q[fa].l=i; q[i].f=fa;
}
else{
int r=fa;
while(q[r].r>-1) r=q[r].r;//找右子树最末端
num++; q[r].r=num; q[num].f=r;
q[num].l=i; q[i].f=num;//新增一个节点
}
}
}
void checktree(int p){
if(p==-1) return;
cout<<p<<' '<<q[p].f<<' '<<q[p].w<<' '<<q[p].j<<endl;
checktree(q[p].l); checktree(q[p].r);
}
void dfs(int p){
int Min=1e+9;
if(p<0) return;
if(p>0){
q[p].deep=q[q[p].f].deep+1;
q[p].d[0]=0;//到自己的距离
for(int i=1;i<=q[p].deep;i++) q[p].d[i]=q[q[p].f].d[i-1]+q[p].j;
}
dfs(q[p].l); dfs(q[p].r);
if(q[p].l<0){
for(int j=0;j<=q[p].deep;j++) dp[p][j][0]=q[p].w*q[p].d[j];
return;
}
for(int i=0;i<=k;i++)
for(int j=0;j<=q[p].deep;j++){
if(i>0) Min=dp[p][0][i-1];
for(int o=0;o<=i;o++) Min=min(Min,dp[q[p].l][j+1][o]+dp[q[p].r][j+1][i-o]+q[p].w*q[p].d[j]);
dp[p][i][j]=Min;
}
}
int main(){
loading();
//checktree(0);//这个树太坤坤健康了
dfs(0);
cout<<dp[0][0][k];
}