【模板】Dijkstra+堆优化

Sirius墨狼 · 2019-11-12 21:52:15 · 个人记录

Dijkstra+堆优化

P4779 【模板】单源最短路径（标准版）

常规写法（722ms）

#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
inline void read(int &x)
{
    char c=0;x=0;
    while(!isdigit(c))c=getchar();
    while(isdigit(c))x=x*10+c-'0',c=getchar();
}
struct edge
{
    int v,w,nxt;
}e[500010];
int head[100010],cnt=0;
inline void addEdge(int u,int v,int w)
{
    e[++cnt].v=v;
    e[cnt].w=w;
    e[cnt].nxt=head[u];
    head[u]=cnt;
}
int n,m,s;
int dis[100010];
struct node
{
    int u,d;
    bool operator <(const node &x)const
    {
        return x.d<u;
    }
};
inline void dijkstra()
{
    for(int i=1;i<=n;i++)
    dis[i]=0x7fffffff;
    dis[s]=0;
    priority_queue<node> Q;
    Q.push((node){s,0});
    while(!Q.empty())
    {
        node fr=Q.top();
        Q.pop();
        int u=fr.u,d=fr.d;
        if(d!=dis[u])continue;
        for(int i=head[u];i;i=e[i].nxt)
        {
            int v=e[i].v,w=e[i].w;
            if(dis[u]+w<dis[v])
            {
                dis[v]=dis[u]+w;
                Q.push((node){v,dis[v]});
            }
        }
    }
}
int main()
{
    int x,y,z;
    read(n);
    read(m);
    read(s);
    for(int i=1;i<=m;i++)
    {
        read(x);
        read(y);
        read(z);
        addEdge(x,y,z);
    }
    dijkstra();
    for(int i=1;i<=n;i++)
    printf("%d ",dis[i]);
    return 0;
}

整合后的hzwer式模板（214ms）

#include<iostream>
#include<cstdio>
#include<queue>
#include<vector>
#define pa pair<int,int>
using namespace std;
inline void read(int &x)
{
    char c=0;x=0;
    while(!isdigit(c))c=getchar();
    while(isdigit(c))x=x*10+c-'0',c=getchar();
}
struct edge
{
    int to,next,v;
}e[2000005];
int n,m,s,t,cnt,last[250005],d[250005];
bool vis[250005];
void insert(int u,int v,int w)//链表存图 
{
    e[++cnt].to=v;
    e[cnt].next=last[u];
    last[u]=cnt;
    e[cnt].v=w;
}
priority_queue<pa,vector<pa>,greater<pa> >q;//优先队列（大根堆） 
void dijkstra()
{
    for(int i=1;i<=n;i++)
    d[i]=0x7fffffff;//全部取最大 
    d[s]=0;//初始点为0 
    q.push(make_pair(0,s));//make_pair合并两边值 
    while(!q.empty())//堆不为空 
    {
        int now=q.top().second;
        q.pop();
        if(vis[now])continue;
        vis[now]=1;
        for(int i=last[now];i;i=e[i].next)//遍历x的所有出边 
        if(d[now]+e[i].v<d[e[i].to])//每次更新相邻边上的点 
        {
            d[e[i].to]=d[now]+e[i].v;
            q.push(make_pair(d[e[i].to],e[i].to));
        }
    }
}
int main()
{
    int x,y,z;
    read(n);
    read(m);
    read(s);//初始点 
    read(t);//结束点 
    for(int i=1;i<=m;i++)
    {
        read(x);
        read(y);
        read(z);
        insert(x,y,z);
    }
    dijkstra();
    //for(int i=1;i<=n;i++)
    //printf("%d ",d[i]);
    printf("%d\n",d[t]);
    return 0;
}