求助站外题
  • 板块学术版
  • 楼主HANDSOME_FZZ
  • 当前回复0
  • 已保存回复0
  • 发布时间2023/7/14 16:26
  • 上次更新2023/11/3 09:52:34
查看原帖
求助站外题
421758
HANDSOME_FZZ楼主2023/7/14 16:26
题目描述
    给定一棵有向树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];
}
2023/7/14 16:26
加载中...