先上一波题目qwq https://www.luogu.org/problem/P1629·

复习了一波 dijstra 的 priority_queue(优先队列)优化的写法

tips: 求单项路中求每个点走到固定点的最短路可以把每条路倒过来 然后从固定点开始跑一次最短路即可

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
const int M=10007,mx=2e7;
int read(){
    int ans=0,f=1,c=getchar();
    while(c<'0'||c>'9'){if(c=='-') f=-1; c=getchar();}
    while(c>='0'&&c<='9'){ans=ans*10+(c-'0'); c=getchar();}
    return ans*f;
}
struct qwq{
    int id,d;
    bool operator<(const qwq&x)const{return x.d<d;}
};
priority_queue<qwq>q;
int cnt,first[M];
struct node{int to,next,w;}e[100007],e1[100007];
void ins(int a,int b,int w){
    cnt++;      e[cnt].to=b; e[cnt].w=w;
    e[cnt].next=first[a]; first[a]=cnt;
}
int cnt1,first1[M];
void ins1(int a,int b,int w){
    cnt1++; e1[cnt1].to=b; e1[cnt1].w=w;
    e1[cnt1].next=first1[a]; first1[a]=cnt1;
}
int n,m,dis[M],dis1[M];
void dj1(){
    q.push((qwq){1,0});
    dis[1]=0;
    while(!q.empty()){
        qwq x=q.top(); q.pop();
        if(x.d>dis[x.id]) continue;
        for(int i=first[x.id];i;i=e[i].next){
            int now=e[i].to;
            if(dis[now]>dis[x.id]+e[i].w){
                dis[now]=dis[x.id]+e[i].w;
                q.push((qwq){now,dis[now]});
            }
        }
    }
}
void dj2(){
    q.push((qwq){1,0});
    dis1[1]=0;
    while(!q.empty()){
        qwq x=q.top(); q.pop();
        if(x.d>dis1[x.id]) continue;
        for(int i=first1[x.id];i;i=e1[i].next){
            int now=e1[i].to;
            if(dis1[now]>dis1[x.id]+e1[i].w){
                dis1[now]=dis1[x.id]+e1[i].w;
                q.push((qwq){now,dis1[now]});
            }
        }
    }
}
int main(){
    int x,y,w,ans=0;
    n=read(); m=read();
    for(int i=1;i<=n;i++) dis[i]=mx,dis1[i]=mx;
    for(int i=1;i<=m;i++) x=read(),y=read(),w=read(),ins(x,y,w),ins1(y,x,w);
    dj1(); dj2();
    for(int i=2;i<=n;i++) ans+=dis[i]+dis1[i];
    printf("%d\n",ans);
    return 0;
}
View Code
02-01 17:24