洛谷:P1629 邮递员送信 dijkstra：求所有点到起点的距离

题目：

题解：

正常建图用dijkstra算法求从起点出发到所有终点的最短距离。

反向建图用dijkstra算法求从所有终点出发到起点的最短距离。

反向建图：

将正常图的边的方向反向边权正常存储。

正常建图的dikstra算法和反向建图的dijktra算法的操作时一模一样的，只是dist的含义变了：

正常建图：dist[ i ] 代表从起点到顶点 i 的最短距离。

反向建图：dist[ i ] 代表从终点 i 到起点的最短距离。

代码：

#include<bits/stdc++.h>
using namespace std;
using PII=pair<int,int>;

const int N=1010,M=100010;
int n,m;

int h[N],e[M],ne[M],w[M],idx;
int h1[N],e1[M],ne1[M],w1[M],idx1;


int dist[N];
int st[N];

//反向建图的dijkstra算法与正常建图的dijkstra算法的函数是一模一样的，只不过使用的边有所不同
void dijkstra(int s,int h[],int w[],int e[],int ne[]){
    memset(dist,0x3f, sizeof dist);
    memset(st,0,sizeof st);
    priority_queue<PII,vector<PII>,greater<PII> >heap;
    dist[s]=0;
    heap.push({0,s});
    while(heap.size()){
        auto  t=heap.top();
        heap.pop();
        int p=t.second,distance=t.first;
        if(st[p])continue;
        st[p]=true;
        for(int i=h[p];i!=-1;i=ne[i]){
            int j=e[i];
            if(dist[j]>dist[p]+w[i]){
                dist[j]=dist[p]+w[i];
                heap.push({dist[j],j});
            }
        }
    }


    
}

void add1(int a,int b,int c){
    e1[idx1]=b,w1[idx1]=c,ne1[idx1]=h1[a],h1[a]=idx1++;
}
void add(int a,int b,int c){
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}

int main(){
    scanf("%d%d",&n,&m);
    memset(h,-1,sizeof h);
    memset(h1,-1,sizeof h1);
    for(int i=0;i<m;i++){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
        add1(b,a,c);
    }
    int ans=0;
    dijkstra(1,h1,w1,e1,ne1);
    for(int i=2;i<=n;i++){
        ans+=dist[i];
    }
    dijkstra(1,h,w,e,ne);
    for(int i=2;i<=n;i++){
        ans+=dist[i];
    }
   
    printf("%d",ans);
    return 0;
}

题后反思：

dijkstra算法既可以求从一个顶点出发到所有终点的最短距离，也可以求从所有终点出发到起点的最短距离。