1. 程式人生 > >HDU 2544 最短路【Dijkstra演算法堆優化,Vector建圖】

HDU 2544 最短路【Dijkstra演算法堆優化,Vector建圖】

最短路

Time Limit: 5000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 55757    Accepted Submission(s): 24651


Problem Description 在每年的校賽裡,所有進入決賽的同學都會獲得一件很漂亮的t-shirt。但是每當我們的工作人員把上百件的衣服從商店運回到賽場的時候,卻是非常累的!所以現在他們想要尋找最短的從商店到賽場的路線,你可以幫助他們嗎?


Input 輸入包括多組資料。每組資料第一行是兩個整數N、M(N<=100,M<=10000),N表示成都的大街上有幾個路口,標號為1的路口是商店所在地,標號為N的路口是賽場所在地,M則表示在成都有幾條路。N=M=0表示輸入結束。接下來M行,每行包括3個整數A,B,C(1<=A,B<=N,1<=C<=1000),表示在路口A與路口B之間有一條路,我們的工作人員需要C分鐘的時間走過這條路。
輸入保證至少存在1條商店到賽場的路線。

Output 對於每組輸入,輸出一行,表示工作人員從商店走到賽場的最短時間
Sample Input 2 1 1 2 3 3 3 1 2 5 2 3 5 3 1 2 0 0
Sample Output 3 2
Source

AC程式碼:

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
const int maxn=100+10;
#define INF 1e9
int n,m;
struct Edge
{
    int u,v,w;
    Edge(int u,int v,int w):u(u),v(v),w(w) {}
};

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

vector<Edge> edges;
vector<int> G[maxn];
bool done[maxn];
int d[maxn];

void init()
{
    for(int i=0; i<n; i++)
        G[i].clear();
    edges.clear();
}

void AddEdge(int u,int v,int w)
{
    edges.push_back(Edge(u,v,w));
    int mm=edges.size();
    G[u].push_back(mm-1);
}

void dijkstra()
{
    priority_queue<Node> Q;
    for(int i=0; i<=n; i++) d[i]=INF;
    d[0]=0;
    memset(done,0,sizeof(done));
    Q.push(Node(d[0],0));

    while(!Q.empty())
    {
        Node 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.v] > d[u]+e.w)
            {
                d[e.v] = d[u]+e.w;
                Q.push(Node(d[e.v],e.v));
            }
        }
    }
}

int main()
{
    //freopen("2544.txt","r",stdin);
    while(scanf("%d%d",&n,&m)==2)
    {
        if(n==0 && m==0) break;
        init();
        for(int i=0; i<m; i++)
        {
            int u,v,d;
            scanf("%d%d%d",&u,&v,&d);
            u--,v--;
            AddEdge(u,v,d);
            AddEdge(v,u,d);
        }
        dijkstra();
        printf("%d\n",d[n-1]);
    }
    return 0;
}