201609-4 交通规划

问题描述

　　G国国王来中国参观后，被中国的高速铁路深深的震撼，决定为自己的国家也建设一个高速铁路系统。
　　建设高速铁路投入非常大，为了节约建设成本，G国国王决定不新建铁路，而是将已有的铁路改造成高速铁路。现在，请你为G国国王提供一个方案，将现有的一部分铁路改造成高速铁路，使得任何两个城市间都可以通过高速铁路到达，而且从所有城市乘坐高速铁路到首都的最短路程和原来一样长。请你告诉G国国王在这些条件下最少要改造多长的铁路。

输入格式

　　输入的第一行包含两个整数n, m，分别表示G国城市的数量和城市间铁路的数量。所有的城市由1到n编号，首都为1号。
　　接下来m行，每行三个整数a, b, c，表示城市a和城市b之间有一条长度为c的双向铁路。这条铁路不会经过a和b以外的城市。

输出格式

　　输出一行，表示在满足条件的情况下最少要改造的铁路长度。

样例输入

4 5
1 2 4
1 3 5
2 3 2
2 4 3
3 4 2

样例输出

11

评测用例规模与约定

　　对于20%的评测用例，1 ≤ n ≤ 10，1 ≤ m ≤ 50；
　　对于50%的评测用例，1 ≤ n ≤ 100，1 ≤ m ≤ 5000；
　　对于80%的评测用例，1 ≤ n ≤ 1000，1 ≤ m ≤ 50000；

　　对于100%的评测用例，1 ≤ n ≤ 10000，1 ≤ m ≤ 100000，1 ≤ a, b ≤ n，1 ≤ c ≤ 1000。输入保证每个城市都可以通过铁路达到首都。

#include<iostream>
#include<vector>
#include<queue>
#include<memory.h>
#define INF 9999
#define maxn 10001
using namespace std;

struct Edge{
	int from,to,dist;
	Edge(int u,int v,int d):from(u),to(v),dist(d){}
}; 

struct HeapNode{
	int d,u;
	bool operator < (const HeapNode& rhs) const{
		return d > rhs.d;
	}
};

struct Dijkstra{
	int n,m;
	vector<Edge> edges;
	vector<int> G[maxn];
	bool done[maxn];
	int d[maxn];
	int p[maxn];
	int cost[maxn];
	
	void init(int n){
		this->n=n;
		for(int i=0;i<=n;i++) G[i].clear();//!!!!!!!!!!!!!!!!!!!! 
		edges.clear();
	}
	
	void AddEdge(int from,int to,int dist){
		edges.push_back(Edge(from,to,dist));
		m=edges.size();
		G[from].push_back(m-1);//存储边的编号 
	}
	int ne(){
		return edges.size();
	}
	void dijkstra(int s){
//		int sum=0;
		priority_queue<HeapNode> Q;
		for(int i=0;i<=n;i++) d[i] = INF;//!!!!!!!!!!!!!!!!!!!!!!
		d[s]=0;
		memset(done,0,sizeof(done));
		Q.push((HeapNode){0,s});
		while(!Q.empty()){
			HeapNode x = Q.top();
			Q.pop();
			int u =x.u;
			
			if(done[u]) continue;
			done[u]=true;
			for(int i=0;i<G[u].size();i++){
				Edge& e = edges[G[u][i]];
				if(d[e.to]>d[u]+e.dist){
//					//11111111111 start
//					if(d[e.to]!=INF){
//						Edge& e1 = edges[p[e.to]];
//						sum-=e1.dist;
//					}
//					//111111111111 end
					d[e.to]=d[u]+e.dist;
					p[e.to]=G[u][i];
				//	sum+=e.dist;//111111111111111
					cost[e.to]=e.dist;//2222222222222222
					Q.push((HeapNode){d[e.to],e.to});
				}
				else if(d[e.to]==(d[u]+e.dist)){/

//					Edge& e2 = edges[p[e.to]];
//					if(e2.dist>e.dist){
//						sum-=e2.dist;
//						sum+=e.dist;
//					}
					if(e.dist<cost[e.to]){
						cost[e.to]=e.dist;
						p[e.to]=G[u][i];
					}
				}
			}
		}

	}
	int summ(){
		int sum=0;
		for(int i=1;i<=n;i++){//!!!!!!!!!!!!!
			sum+=cost[i];
		}
		return sum;
	}

};
int main(){
	int n,e;
	cin>>n>>e;
	
	Dijkstra dij;
	dij.init(n);
	for(int i=0;i<e;i++){
		int a,b,c;
		cin>>a>>b>>c;
		dij.AddEdge(a,b,c);
		dij.AddEdge(b,a,c);
	}

	dij.dijkstra(1);

	int sum=dij.summ();
	
	cout<<sum<<endl;
}