POJ - 2387 Til the Cows Come Home(dijkstra)

题意:N和M(0<N<200,0<M<1000),分别代表现有已修建的道路的数目和城镇的数目,城镇分别以0~M-1编号。接下来N条道路 A, B, X  (0<=A,B<N,A!=B,0<X<10000) ,

两个整数S,T(0<=S,T<N),分别代表起点和终点。如果数据不存在则输出-1

思路:使用优先队列优化的迪杰斯特拉算法,由于输入中还有重边问题,用链式向前星(没有处理的情况下)就会有错,所以我们用vector存储的邻接表 来存储

 

完整代码:


#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <queue>
#include <algorithm>
#define inf 0x3f3f3f3f 
#define maxn 1010
using namespace std;
int dist[maxn],n,t;
struct edge
{
    int to;
    int cost;
};
vector<edge> g[maxn];//邻接表 
typedef pair<int,int> pi;

void dijkstra(int s)
{
    //通过指定greater<pi>参数,堆按照first从大到小的顺序取初值
    priority_queue<pi,vector<pi>,greater<pi> > que;
   	memset(dist,inf,sizeof(dist));
    dist[s]=0;
    que.push(pi(0,s));
    while(!que.empty())
    {
        pi p = que.top();
        que.pop();
        int v = p.second;//first是最短距离,second是顶点的编号
        if(dist[v]<p.first) continue;
        
        for(int i=0; i<g[v].size(); ++i)
        {
            edge e=g[v][i];
            if(dist[e.to]>dist[v]+e.cost)//Dijkstra
            {
                dist[e.to]=dist[v]+e.cost;
                que.push(pi(dist[e.to],e.to));
            }
        }
    }
}
int main()
{
    int a,b,l;
    while(scanf("%d%d",&t,&n)!=EOF)
    {
        for(int i=0; i<t; i++)
        {
            scanf("%d%d%d",&a,&b,&l);
            edge A; A.cost = l,A.to = b;
            edge B; B.cost = l,B.to = a;
            g[a].push_back(A);
            g[b].push_back(B);
        }
        dijkstra(n);
        printf("%d\n",dist[1]);
    }
    return 0;
}
 

 

posted @ 2019-08-01 13:35  Tianwell  阅读(115)  评论(0编辑  收藏  举报