HDU 1874 畅通工程续(dijkstra+优先队列)

畅通工程续

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

 

#include<cstdio>
#include<queue>
#include<algorithm>
#include<cstring>
using namespace std;

struct edge
{
    int from,to,dist;
    edge(int u,int v,int w):from(u),to(v),dist(w){};
};

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

const int inf=0x3f3f3f3f;
int m,n;
vector<edge>edges;
vector<int>G[205]; 
int vis[205],d[205];

void init()
{
    for(int i=0;i<m;i++)
    G[i].clear();
    edges.clear();
    memset(vis,0,sizeof(vis));
}

void addedge(int u,int v,int w)
{
    edges.push_back(edge(u,v,w));
    edges.push_back(edge(v,u,w));
    int m=edges.size();
    G[u].push_back(m-2); 
    G[v].push_back(m-1);
}

int dijkstra(int S,int T)
{
    priority_queue<heapnode>q;
    q.push((heapnode){S,0});
    for(int i=0;i<m;i++)d[i]=inf;
    d[S]=0;
    while(!q.empty())
    {
        int u=q.top().u;
        q.pop();
        if(vis[u])
        continue;
        vis[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)
            {
                d[e.to]=d[u]+e.dist;
                q.push((heapnode){e.to,d[e.to]});
            }
        }
    }
    return d[T];
}

int main()
{
    //freopen("in.txt","r",stdin);
    int a,b,c,S,T;
    while(scanf("%d%d",&m,&n)!=EOF)
    {
        init();
        for(int i=0;i<n;i++)
        {
            scanf("%d%d%d",&a,&b,&c);
            addedge(a,b,c);
        }
        scanf("%d%d",&S,&T);
        int ans=dijkstra(S,T);
        if(ans==inf)
        printf("-1\n");
        else
        printf("%d\n",ans); 
    }
    return 0;
}