dijstra优化(优先队列+邻接表)

以poj2387为例

dijstra的思想是每次在dis数组里面找出最小的那一条边,然后用这一条边去松弛其他的边,在这里我们把dis数组保存在一个优先队列里面,同时把图存在邻接表里面,每次通过优先队列得出dis数组里面最小的那条边,然后通过邻接表去松弛与这条边相连的其他的边,然后把更新的dis数组存入优先对列,直到队列为空时结束。

ac代码:

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
const int inf=0x3f3f3f3f;
const int MOD=1e9+7;
const int inff=999999;
const int N=1005;
const int M=2005;
#define LL long long
#define ME0(x) memset(x,0,sizeof(x))
#define MEF(x) memset(x,-1,sizeof(x))
#define MEI(x) memset(x,inf,sizeof(x))
using namespace std;
int n,m,cnt;
struct LX
{
    int ed,di,next;
}lx[M*2];
int first[N];
int dis[N],vis[N];
void add(int s,int e,int d)
{
    cnt++;
    lx[cnt].ed=e;
    lx[cnt].di=d;
    lx[cnt].next=first[s];
    first[s]=cnt;
}
struct cmp
{
    bool operator ()(const int a,const int b)
    {
        return dis[a]>dis[b];
    }
};
int dijstra(int s)
{
    priority_queue,cmp> pq;
    for(int n1=1;n1<=n;n1++)
    {
        dis[n1]=inff;
    }
    dis[s]=0;
    ME0(vis);
    pq.push(s);
    while(!pq.empty())
    {
        int xe=pq.top();
        pq.pop();
        if(vis[xe])
        {
            continue;
        }
        vis[xe]=1;
        for(int i=first[xe];i;i=lx[i].next)
        {
            int ie=lx[i].ed;
            int di=lx[i].di;
            if(!vis[ie]&&dis[ie]>dis[xe]+di)
            {
                dis[ie]=dis[xe]+di;
                pq.push(ie);
            }
        }
    }
    return dis[n];
}
int main()
{
    int s,e,d;
    scanf("%d%d",&m,&n);
    cnt=0;
    ME0(first);
    for(int m1=1;m1<=m;m1++)
    {
        scanf("%d%d%d",&s,&e,&d);
        add(s,e,d);
        add(e,s,d);
    }
    printf("%d\n",dijstra(1));
}

 

你可能感兴趣的:(STL,最短路)