【图论－－Dijkstra】HDU 1874 畅通工程续

HDU 1874 畅通工程续                                                                     

畅通工程续

Time Limit: 3000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 62484    Accepted Submission(s): 23498
题目链接－>http://acm.hdu.edu.cn/showproblem.php?pid=1874

Problem Description

某省自从实行了很多年的畅通工程计划后，终于修建了很多路。不过路多了也不好，每次要从一个城镇到另一个城镇时，都有许多种道路方案可以选择，而某些方案要比另一些方案行走的距离要短很多。这让行人很困扰。

现在，已知起点和终点，请你计算出要从起点到终点，最短需要行走多少距离。

 

Input

本题目包含多组数据，请处理到文件结束。
每组数据第一行包含两个正整数N和M(0<N<200,0<M<1000)，分别代表现有城镇的数目和已修建的道路的数目。城镇分别以0～N-1编号。
接下来是M行道路信息。每一行有三个整数A,B,X(0<=A,B<N,A!=B,0<X<10000),表示城镇A和城镇B之间有一条长度为X的双向道路。
再接下一行有两个整数S,T(0<=S,T<N)，分别代表起点和终点。

 

Output

对于每组数据，请在一行里输出最短需要行走的距离。如果不存在从S到T的路线，就输出-1.

 

Sample Input

3 3
0 1 1
0 2 3
1 2 1
0 2
3 1
0 1 1
1 2

Sample Output

2
-1

  Problem Idea

　【题意】

    题意为：输入一个节点数为n，边数为m的无向图的各条边，共m条边的信息。且需要自己给定起点和终点。

　　要求你输出从起点节点s到终点节点e的距离的最大值:

      若DJ.dijkstra(int s,int e)方法返回为void,只需求DJ.d[e]

      若DJ.dijkstra(int s,int e)方法返回为int,只需得到while循环结束后返回的d[e]

　　注意：q.push(HeapNode(d[s],s));本题中优先队列初始化的节点为HeapNode(d[s],s)

　　区别：注意区别与求解从第１号节点到所有其他点的距离的最大值

　【类型】

　　Dijkstra算法求单源最短路径

　【分析】

　　模板题，入门

　【时间复杂度&&优化】

　　O(nlog n)

 Souce Code

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

const int nmax=200+10; //nmax取值看题目给的n的范围
#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的最短路中的最后一条边的编号
    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开始)
    }

    int dijkstra(int s,int e){//求s到所有点的距离
        //返回值int类型的d[e]为:源点s到终点e的最短路径d[e]
        for(int i=0;i<n;i++) d[i]=INF;
        d[s]=0;//记得源点从0开始,不是1开始
        memset(done,0, sizeof(done));

        priority_queue<HeapNode>q;
        q.push(HeapNode(d[s],s));//本题中为d[s],s

        while(!q.empty()){
            //在所有未标号的节点中，选出d值最小的节点
            HeapNode x=q.top();//找到d值最大的节点
            q.pop();//弹出队头
            int u=x.u;//得到d值最大的节点的起点
            //给节点x标记
            if(done[u]) continue; //如果当前节点已经标号，则continue
            done[u]=true; //若未标号，则标记
            //遍历从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
                }
            }
        }
        //这段代码要加到while外面
        if(d[e]==INF){
            return -1;
        }
        return d[e];//返回源点s到终点e的最短路径d[e]
    }
}DJ;
int main() {
    while(cin>>n>>m){//输入节点数
        DJ.init(n);//切记要初始化，清空边列表和邻接表
        while(m--) {
            int u,v,d;
            scanf("%d%d%d",&u,&v,&d);//输入节点是从0开始的,看题目要求城镇是从0到N-1编号的
            //u--;v--;
            DJ.AddEdge(u,v,d);
            DJ.AddEdge(v,u,d);
        }
        int s,e;
        cin>>s>>e;
        int len=DJ.dijkstra(s,e);//得到节点s到节点e的最短路径长度

        cout<<len<<endl;//求节点s到节点e的最短距离

    }
    return 0;
}