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

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

题目背景

2018 年 7 月 19 日，某位同学在 NOI Day 1 T1 归程 一题里非常熟练地使用了一个广为人知的算法求最短路。

然后呢？

100→60；

Ag→Cu；

最终，他因此没能与理想的大学达成契约。

小 F 衷心祝愿大家不再重蹈覆辙。

题目描述

给定一个 N 个点，M 条有向边的带非负权图，请你计算从 S 出发，到每个点的距离。

数据保证你能从 S 出发到任意点。

---

Dijkstra

定义

Dijkstra(迪杰斯特拉)算法是典型的单源最短路径算法，用于计算一个节点到其他所有节点的最短路径。主要特点是以起始点为中心向外层层扩展，直到扩展到终点为止

实现

算法思想：设G=(V,E)是一个带权有向图，把图中顶点集合V分成两组，第一组为已求出最短路径的顶点集合（用S表示，初始时S中只有一个源点，以后每求得一条最短路径 , 就将加入到集合S中，直到全部顶点都加入到S中，算法就结束了），第二组为其余未确定最短路径的顶点集合（用U表示），按最短路径长度的递增次序依次把第二组的顶点加入S中。在加入的过程中，总保持从源点v到S中各顶点的最短路径长度不大于从源点v到U中任何顶点的最短路径长度。此外，每个顶点对应一个距离，S中的顶点的距离就是从v到此顶点的最短路径长度，U中的顶点的距离，是从v到此顶点只包括S中的顶点为中间顶点的当前最短路径长度。

算法步骤：
a.初始时，S只包含源点，即S＝{v}，v的距离为0。U包含除v外的其他顶点，即:U={其余顶点}，若v与U中顶点u有边，则<u,v>正常有权值，若u不是v的出边邻接点，则<u,v>权值为∞。
b.从U中选取一个距离v最小的顶点k，把k，加入S中（该选定的距离就是v到k的最短路径长度）。
c.以k为新考虑的中间点，修改U中各顶点的距离；若从源点v到顶点u的距离（经过顶点k）比原来距离（不经过顶点k）短，则修改顶点u的距离值，修改后的距离值的顶点k的距离加上边上的权。
d.重复步骤b和c直到所有顶点都包含在S中。

主要是借助优先队列实现
pair默认以first优先排序 保证取出来的点与源点最近
一开始不知道为什么要用结构体QAQ

不加优化TLE

#include <iostream>
#include <cstdio>
#include <queue>
#define INF 0x3f3f3f3f
#define ll long long
using namespace std;
const ll N=100005;
const ll M=200005;
ll n,m,s;
ll u,v,w;
ll cnt=0;
ll head[N],dis[N],vis[N];
struct Edge
{
    ll to,next,dis;
}e[M];
priority_queue<pair<int,int>,vector< pair<int,int> >,greater<pair<int,int> > > q;
inline ll read()
{
    char c;
    ll x=0,f=1;
    do{
        c=getchar();
        if(c=='-')
            f=-1;
    }while(c<'0' || c>'9');
    do{
        x=(x<<3)+(x<<1)+(c^48);
        c=getchar();
    }while(c>='0' && c<='9');
    return f*x;
}
inline void addedge(int u,int v,int w)
{
    cnt++;
    e[cnt].to=v;
    e[cnt].dis=w;
    e[cnt].next=head[u];
    head[u]=cnt;
}
inline void in()
{
    for(int i=1;i<=n;++i)
    {
        head[i]=-1;
        vis[i]=0;
        dis[i]=INF;
    }
    dis[s]=0;
}
inline void dijkstra()
{
    q.push(make_pair(0,s));
    while(!q.empty())
    {
        int x=q.top().second;
        q.pop();
        if(vis[x])
            continue;
        //优化1 同一个点在被标记前可能被push了多次
        vis[x]=1;
        int h=head[x];
        while(h>0)
        {
            int y=e[h].to;
            if(dis[x]+e[h].dis<dis[y])
            {
                dis[y]=dis[x]+e[h].dis;
                if(!vis[y])
                {
                    q.push(make_pair(dis[y],y));
                }
            }
            h=e[h].next;
        }
    }
}
int main(){
    n=read(),m=read(),s=read();
    in();
    for(int i=0;i<m;++i)
    {
        u=read(),v=read(),w=read();
        if(u==v)
            continue;
        addedge(u,v,w);
    }
    dijkstra();
    for(int i=1;i<=n;++i)
    {
        printf("%lld ",dis[i]);
    }
    printf("\n");
    return 0;
}

#include <iostream>
#include <cstdio>
#include <queue>
#define INF 0x3f3f3f3f
#define ll long long
using namespace std;
const ll N=100005;
const ll M=200005;
ll n,m,s;
ll u,v,w;
ll cnt=0;
ll head[N],dis[N],vis[N];
struct Edge
{
    ll to,next,dis;
}e[M];
priority_queue<pair<int,int>,vector< pair<int,int> >,greater<pair<int,int> > > q;
inline ll read()
{
    char c;
    ll x=0,f=1;
    do{
        c=getchar();
        if(c=='-')
            f=-1;
    }while(c<'0' || c>'9');
    do{
        x=(x<<3)+(x<<1)+(c^48);
        c=getchar();
    }while(c>='0' && c<='9');
    return f*x;
}
inline void addedge(int u,int v,int w)
{
    cnt++;
    e[cnt].to=v;
    e[cnt].dis=w;
    e[cnt].next=head[u];
    head[u]=cnt;
}
inline void in()
{
    for(int i=1;i<=n;++i)
    {
        head[i]=-1;
        vis[i]=0;
        dis[i]=INF;
    }
    dis[s]=0;
}
inline void dijkstra()
{
    q.push(make_pair(0,s));
    while(!q.empty())
    {
        int x=q.top().second;
        if(dis[x]!=q.top().first)//优化2 说明同一个点被更新且push了多次
        {
            q.pop();
            continue;
        }
        vis[x]=1;
        q.pop();
        int h=head[x];
        while(h>0)
        {
            int y=e[h].to;
            if(dis[x]+e[h].dis<dis[y])
            {
                dis[y]=dis[x]+e[h].dis;
                if(!vis[y])
                    q.push(make_pair(dis[y],y));
            }
            h=e[h].next;
        }
    }
}
int main(){
    n=read(),m=read(),s=read();
    in();
    for(int i=0;i<m;++i)
    {
        u=read(),v=read(),w=read();
        if(u==v)
            continue;
        addedge(u,v,w);
    }
    dijkstra();
    for(int i=1;i<=n;++i)
    {
        printf("%lld ",dis[i]);
    }
    printf("\n");
    return 0;
}

两种优化针对的情况是一样的
只是理解和应对方法不同

感谢@Wings和@万弘提供的优化