插排树(不是dfs喔)

题目链接:https://www.nowcoder.com/acm/contest/157/C

参考链接:https://www.nowcoder.com/acm/contest/view-submission?submissionId=31648555

链接:https://www.nowcoder.com/acm/contest/157/D
来源:牛客网

插排序

一年一度的山东省oi夏令营又开始了,每到这个季节,山东的oier们都会欢聚这里,一起学(tuí)习(feì)。当然,为了能更加愉快地学(tuí)习(feì),就少不了要自带电脑,用电便开始成了一种问题,于是便有一种神奇的数据结构诞生了!这就是山东省oi专用数据结构——插排树(如图)

小K为了能更好的学(tuí)习(feì),所以他想尽量的往后做,所以现在请你帮帮他,他最远可以离讲台多远。

已知插排树的根节点在讲台上,有且仅有一个根节点(根节点入度为0),最远距离即所有插排的长度,小K电脑线的长度忽略不计

本题良心大样例下载地址: https://kench.co/tree.zip

输入描述:

第一行一个整数n表示有n个节点
然后n-1行,每行三个整数a,b,c,表示插排a是接在插排b上的,插排a的长度为c

输出描述:

一个整数n表示最远距离

示例1

输入

复制

9
2 1 2
3 1 2
4 1 1
5 2 3
6 2 1
7 3 1
8 3 4
9 7 5

输出

复制

8

说明

1=>3=>7=>9

备注:

对于30%的数据 n<233
对于70%的数据 n<2333

对于100%的数据 n<50000

c小于20

a,b小于等于n

我的天呐,猛一看这不是dfs吗,再看一下数据范围,n<50000,蒙了,2 的50000此方是多少来着……装不下,装不下,

其实我们可以这样来,通过把权值往孩子节点传递,权值依次往下累加,最后再来一次循环,孩子节点权值最大的就是最终ans

#include<iostream>
#include<algorithm>
#include<cstring> 
using namespace std;

const int MAXN = 50000+10;

struct node{
	int a;
	int b;
	int c;
}vec[MAXN];
int ans[MAXN];
bool cmp(node x,node y){
	return x.a<y.a;
}
int main(){
	
	memset(ans,0,sizeof ans);
	memset(vec,0,sizeof vec);
	
	int n;
	cin>>n;
	n-=1;
	for(int i=1;i<=n;i++){
		cin>>vec[i].a>>vec[i].b>>vec[i].c;
	}
	sort(vec+1,vec+n+1,cmp);
	int e;
	int f;
	int g;
	for(int i=1;i<=n;i++){
		e=vec[i].a;
		f=vec[i].b;
		g=vec[i].c;
		ans[e]=ans[f]+g; // 核心代码; 
	}
	int sum=ans[1];
	//i<=n+1,是因为 n 之前自减 1 但是节点中有编号 n; 
	for(int i=1;i<=n+1;i++){
		sum=max(sum,ans[i]);
	}
	cout<<sum<<endl;
	return 0;
} 

猜你喜欢

转载自blog.csdn.net/guozuofeng/article/details/81668337