【CCF 201609-4】 交通规划 （图论－－Dijkstra）

CCF 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。输入保证每个城市都可以通过铁路达到首都。

 

 　【题意】

  　“所有城市乘坐高速铁路到首都的最短路程和原来一样长”，说明结果满足单源最短路径；“最少要改造多少铁路”，说明是要在最短路径中找最小花费。如果同时存在多条最短路径，应该选择扩展时用到的花费最小的那一条。
　【举例说明】

　　如下图的例子所示，点1到点3的最短路径是4，要连通点3，1-2-3、1-3和1-4-3都是最短路径。但是如果选1-3，需要增加的铁轨为4个单位；选1-2-3需要增加的铁轨为2个单位；而选1-4-3的话，需要增加的铁轨只有1个单位。所以此时应该选最后一种方案。

　【类型】
　　Dijkstra算法＋贪心法
　【分析】
　　可套用Dijkstra算法，但是需要做一些改进。本题中我们用Dijkstra算法不是要得到最短路径，而是要得到最短路径下连通每个点所增加的最小的边是多少，即在最短路径中找最小花费。如果用cost[v]表示连通v点所增加的边的权重，比如上图中cost[3]=1。当遇到上述多种选项时，也就是dist[v]==disto[u]+cost时，让cost[v]=min(cost[v],cost)，这样最终得到的cost[v]就是满足最短路径条件下的最小花费。

int cost[nmax]; //接通该点需要增加的边的权重

//遍历从x出发的所有边(x,y),更新d[y]=min{d[y],d[x]+w[x,y]}
            for(int i=0;i<G[u].size();i++){
                Edge& e=edges[G[u][i]];//得到从u起点出发到i节点的一条边e

                if(d[e.to]>d[u]+e.dist){
                    d[e.to]=d[u]+e.dist;
                    //p[e.to]=G[u][i];
                    q.push(HeapNode(d[e.to],e.to));//添加节点y

                    cost[e.to]=e.dist;//更新连通e.to点所增加的边的权重为e.dist
                }
                //记录下边权最小的一个
               /*if(d[e.to]==d[u]+e.dist && cost[e.to]>e.dist)//可以这么写
                    cost[e.to]=e.dist;*/
                if(d[e.to]==d[u]+e.dist)
                   cost[e.to]=min(cost[e.to],e.dist);
            }

　【注意】

　　　　１、切记：nmax节点数量的最大值要记得改！！！本题中n的最大值为10000,刚开始模板中const int nmax＝100+5，声明的namx不够大，提交之后会超时的，提交了好多次才发现这个问题！

　　　　　２、本题目中新定义了一个cost数组，记得和他有关的要初始化！

for(int i=0;i<n;i++) {
            d[i]=INF;
            cost[i]=INF;//初始化cost

        }
        d[s]=cost[s]=0;//记得源点从0开始,不是1开始

　【时间复杂度&&优化】
　　O(nlog n)

#include <iostream>
#include <queue>
#include <vector>
#include <cstring>
#include <cstdio>

const int nmax=10000+5;
#define INF 1e8
int n,m;
using namespace std;

struct HeapNode{
    int d,u;//d为s到各个节点的距离,u为起点
    HeapNode(){}
    HeapNode(int d,int u):d(d),u(u){}
    bool operator <(const HeapNode &rhs)const{//自定义greater算子，优先输出d小的节点
        return d>rhs.d;
    }
};
struct Edge{
    int from,to,dist;
    Edge(){}
    Edge(int f,int t,int d):from(f),to(t),dist(d){}
};
struct Dijkstra{
    int n,m; //n为点数,m为边数
    vector<Edge> edges;//边列表,存储各边的编号
    vector<int>G[nmax];//邻接表,存储每个节点出发的边编号(从0开始编号)，G[u][i]为起点u到节点i的边的编号
    bool done[nmax];//是否已永久标号
    int d[nmax];//源点s到各个节点的距离,id[i]为源点s到节点i的距离
    int p[nmax];//最短路中的上一条弧,p[i]为源点s到节点i的最短路中的最后一条边的编号

    int cost[nmax]; //接通该点需要增加的边的权重
    Dijkstra(){} //记得写个空的构造函数，要不然DJ会报错

    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){//如果是无向图,每条无向边调用两次AddEdge
       edges.push_back(Edge(from,to,dist));//边列表增加一条边
        m=edges.size();//边列表中有几条边
        G[from].push_back(m-1);//邻接表中的节点数目(下标从0开始)
    }

    void dijkstra(int s){//求源点０到其他节点的最短路径
        for(int i=0;i<n;i++) {
            d[i]=INF;
            cost[i]=INF;//初始化cost

        }
        d[s]=cost[s]=0;//记得源点从0开始,不是1开始
         //本题中，源点为节点０
        memset(done,0, sizeof(done));

        priority_queue<HeapNode>q;
        q.push(HeapNode(d[s],0));//本题中优先队列初始化的节点为HeapNode(d[0],0)

        while(!q.empty()){
            //在所有未标号的节点中，选出d值最小的节点
            HeapNode x=q.top();//找到d值最大的节点
            q.pop();//弹出队头
            int u=x.u;//得到d值最大的节点的起点
            //给节点x标记
            if(done[u]) continue; //如果当前节点已经标号，则continue
            done[u]=1; //若未标号，则标记
            //遍历从x出发的所有边(x,y),更新d[y]=min{d[y],d[x]+w[x,y]}
            for(int i=0;i<G[u].size();i++){
                Edge& e=edges[G[u][i]];//得到从u起点出发到i节点的一条边e

                if(d[e.to]>d[u]+e.dist){
                    d[e.to]=d[u]+e.dist;
                    //p[e.to]=G[u][i];
                    q.push(HeapNode(d[e.to],e.to));//添加节点y

                    cost[e.to]=e.dist;//更新连通e.to点所增加的边的权重为e.dist
                }
                //记录下边权最小的一个
               /*if(d[e.to]==d[u]+e.dist && cost[e.to]>e.dist)
                    cost[e.to]=e.dist;*/
                if(d[e.to]==d[u]+e.dist)
                   cost[e.to]=min(cost[e.to],e.dist);
            }
        }
    }
}DJ;
int main() {
    while(scanf("%d%d",&n,&m)==2){//输入节点数

        DJ.init(n);//切记要初始化，清空边列表和邻接表
        while(m--) {
            int u,v,d;
            scanf("%d%d%d",&u,&v,&d);//输入节点是从1开始的
            u--;v--;//记得把输入节点调整成0开始的
            DJ.AddEdge(u,v,d);
            DJ.AddEdge(v,u,d);
        }

        DJ.dijkstra(0);//得到节点０到其他节点的最短路径长度

        int ans=0;
        for(int i=1;i<=n;i++)
           ans+=DJ.cost[i];

        //cout<<ans<<endl;//求节点0到节点n-1的最短距离
        printf("%d\n",ans);

    }
    return 0;
}