蓝桥杯 大臣的旅费

#include<iostream>
#include<queue>
#define N 10000
using namespace std;
typedef struct Edge {
	int v;
	int info;
	struct Edge* next;
} Edge;
Edge head[N];
int n,sum;
int D[N];
bool visited[N];
queue<int> q;
int bfs_2() {
	int _head=0,_tail=0;
	q.push(1);
	visited[1]=1;
	while(!q.empty()) {
		int v=q.front();
		q.pop();
		for(Edge* i=head[v].next; i; i=i->next) {
			if(!visited[i->v]) {
				D[i->v]=D[v]+(i->info);
				q.push(i->v);
				visited[i->v]=1;
			}
		}
	}
	fill(visited,visited+n+1,0);
	int max=D[1],index=1;
	for(int i=2; i<=n; i++)
		if(max<D[i]) {
			max=D[i];
			index=i;
	}
	fill(D,D+n+1,0);//这里+1是因为是从1开始的

	q.push(index);
	visited[index]=1;
	while(!q.empty()) {
		int v=q.front();
		q.pop();
		for(Edge* i=head[v].next; i; i=i->next) {
			if(!visited[i->v]) {
				D[i->v]=D[v]+(i->info);
				q.push(i->v);
				visited[i->v]=1;
			}
		}
	}
	max=D[1],index=1;
	for(int i=2; i<=n; i++)
		if(max<D[i]) {
			max=D[i];
			index=i;
		}
	
	return max*10+(1+max)*max/2;
}

int main() {
	cin>>n;
	for(int i=0; i<n-1; i++) {
		int x,y,data;
		cin>>x>>y>>data;
		Edge *tp1,*tp2;
		tp1=new Edge;
		tp2=new Edge;
		tp1->info=tp2->info=data;
		tp1->v=y;
		tp1->next=head[x].next;
		head[x].next=tp1;
		tp2->v=x;
		tp2->next=head[y].next;
		head[y].next=tp2;
	}
	cout<<bfs_2();
	return 0;
}