插排树（不是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;
}