概要
问题描述
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。输入保证每个城市都可以通过铁路达到首都。
思路
这是一道Dijikstra与贪心算法结合的题目。利用cost数组保存顶点i最小权重,当dist[v] = dist[vv]+c时,cost[vv]取c与cost[vv]的最小值。
AC代码
#include <iostream>
#include <vector>
#include <queue>
#include <cstring>
using namespace std;
typedef struct edge{
int vertex;
long long cost;
edge(int v,long long c){
vertex = v;
cost = c;
}
}Edge;
struct cmp{
bool operator() (Edge e1,Edge e2){
return e1.cost>e2.cost;
}
};
const long long MAX = 0x7ffffff;
int N,M;
int s,e;
long long c,ans;
vector<Edge> edges[100001];
int collect[100001];
long long dist[100001];
long long cost[100001];
void Dijikstra(int vertex)
{
dist[vertex] = cost[vertex] = 0;
priority_queue<Edge,vector<Edge>,cmp> q;
q.push(Edge(vertex,0));
while(!q.empty()){
Edge tmp = q.top();
q.pop();
int v = tmp.vertex;
if(!collect[v]){
collect[v] = 1;
for(int i = 0 ; i < edges[v].size() ; i++){
int vv = edges[v][i].vertex;
long long c = edges[v][i].cost;
if(collect[vv]){
continue;
}
if(dist[vv] > dist[v]+c){
dist[vv] = dist[v]+c;
cost[vv] = c;
q.push(Edge(vv,dist[vv]));
}
if(dist[vv] == dist[v]+c){
cost[vv] = min(c,cost[vv]);
}
}
}
}
}
int main()
{
while(cin>>N>>M){
ans = 0;
for(int i = 1 ; i <= N ; i++){
collect[i] = 0;
dist[i] = MAX;
cost[i] = MAX;
}
for(int i = 0 ; i < M ; i++){
cin>>s>>e>>c;
edges[s].push_back(Edge(e,c));
edges[e].push_back(Edge(s,c));
}
Dijikstra(1);
for(int i = 1 ; i <= N ; i++){
ans += cost[i];
}
cout<<ans<<endl;
}
return 0;
}