蓝桥杯2013c++A组真题&代码第十题大臣的旅费

蓝桥杯2013c++A组真题&代码第十题大臣的旅费

 这道题主要求的是树的直径，

因为这是无环图，可以看成一棵树，求解思路也很简单，因为题目说了从1作为起点，可以到达任何地方，所以我们从这里开始搜索，找到最远的一个点，然后再根据这个最远的点作为起点，进行搜索，再求得另一个最远的点，这样做就保证只需要进行两次深度优先遍历，就可以求出树的直径。

代码：

代码中，、

dfs 是用了深搜 在这个过程中保存对应的  最远距离和位置

solve 尝试了 dijkstr 算法，求两次最远的距离，结果超时了

/*
标题：大臣的旅费

    很久以前，T王国空前繁荣。为了更好地管理国家，王国修建了大量的快速路，用于连接首都和王国内的各大城市。

    为节省经费，T国的大臣们经过思考，制定了一套优秀的修建方案，使得任何一个大城市都能从首都直接或者
    通过其他大城市间接到达。

    同时，如果不重复经过大城市，从首都到达每个大城市的方案都是唯一的。

    J是T国重要大臣，他巡查于各大城市之间，体察民情。所以，从一个城市马不停蹄地到另一个城市成了J最常做的事情。
    他有一个钱袋，用于存放往来城市间的路费。

    聪明的J发现，如果不在某个城市停下来修整，在连续行进过程中，他所花的路费与他已走过的距离有关，
    在走第x千米到第x+1千米这一千米中（x是整数），
    他花费的路费是x+10这么多。也就是说走1千米花费11，走2千米要花费23。

    J大臣想知道：他从某一个城市出发，中间不休息，到达另一个城市，所有可能花费的路费中最多是多少呢？

输入格式：
输入的第一行包含一个整数n，表示包括首都在内的T王国的城市数
城市从1开始依次编号，1号城市为首都。
接下来n-1行，描述T国的高速路（T国的高速路一定是n-1条）
每行三个整数Pi, Qi, Di，表示城市Pi和城市Qi之间有一条高速路，长度为Di千米。

输出格式:
输出一个整数，表示大臣J最多花费的路费是多少。

样例输入:
5
1 2 2
1 3 1
2 4 5
2 5 4

样例输出:
135

样例说明:
大臣J从城市4到城市5要花费135的路费。


根据资源限制尽可能考虑支持更大的数据规模。


资源约定：
峰值内存消耗 < 64M
CPU消耗  < 5000ms


请严格按要求输出，不要画蛇添足地打印类似：“请您输入...” 的多余内容。

所有代码放在同一个源文件中，调试通过后，拷贝提交该源码。

注意: main函数需要返回0
注意: 只使用ANSI C/ANSI C++ 标准，不要调用依赖于编译环境或操作系统的特殊函数。
注意: 所有依赖的函数必须明确地在源文件中 #include <xxx>， 不能通过工程设置而省略常用头文件。

提交时，注意选择所期望的编译器类型。

 */
//本题求树的直径

#include<iostream>
#include<queue>
#include<cstring>
#include<cmath>
#include<vector>
#include<algorithm>
using namespace std;
typedef long long ll;
const int MAXN = 10000;
int N; 
struct edge{
	int to,cost,crv;
	edge(int to,int cost,int crv):to(to),cost(cost),crv(crv){}
};

vector<edge> map[MAXN];

void add_edge(int from,int to, int cost){
	map[from].push_back(edge(to,cost,map[to].size()));
	map[to].push_back(edge(from,cost,map[from].size()-1));
}

int dist[MAXN];
int used[MAXN];
void solve(int start, int &max_,int &max_index){
	memset(dist,-1,sizeof(dist));
	memset(used,0,sizeof(used));
//	used[start] = 1;
 	dist[start] = 0;
	while(1){
		int v = -1;
		for(int i=0;i<N;i++){
			if(!used[i] && (v==-1 || dist[i]>dist[v])) v = i;
		}
		if(v == -1) break;
		used[v] = 1;
		int length = map[v].size();
		for(int i=0;i<length;i++){
			edge e = map[v][i];
			if(used[e.to]) continue;
			dist[e.to] = max(dist[e.to], dist[v] +e.cost );
			if(dist[e.to] > max_) {
				max_ = dist[e.to];
				max_index = e.to;
			}
		}	
	}
}


void dfs(int start, int parent,int length, int &max_,int &max_index){
	if(length > max_){
		max_ = length;
		max_index = start;
	}

	for(int i=0;i<map[start].size();i++){
		edge e = map[start][i];
		if(e.to==parent)  continue;
		dfs(e.to,start,length+e.cost,max_,max_index);
	}
}


int cal(int num){
	
	int res = 0;
	int temp;
	for(int i=1;i<=num;i++){
		temp = i+10;
		res+=temp;
	}
	return res;
}

int main(){
	scanf("%d",&N);
	for(int i=0;i<N-1;i++){
		int P,Q,D;
		scanf("%d%d%d",&P,&Q,&D);
		add_edge(P-1,Q-1,D);
	}
	
	int max_index = -1;
	int max = -1;
	//从 0 开始出发，找到最远的点，以及对应的距离 
	///dijkstra算法  求解了全部的最大路径实际上没必要 超时 
	//	solve(0,max,max_index);
	//使用深搜，
	dfs(0,-1,0,max,max_index);
	
	int max_index2 = -1;
	int max2 = -1;
	
	//从最远的开始，找另一个最远的点，从而计算最终的树的直径 
	
	///dijkstra算法  求解了全部的最大路径实际上没必要 超时 
//	solve(max_index,max2,max_index2);
	//使用深搜， 
	dfs(max_index,-1,0,max2,max_index2);
	
//	printf("%d %d \n",max_index,max_index2);
	
	printf("%d\n",cal(max2));
	
	return 0;
}