洛谷P1342 请柬(单源最短路dijkstra)

题目

题解

这道题目跟模板差不太多,不同的是它还要求每个点到起点的最短距离,因为我才刚开始学 , 第一时间我想到就是把每个点都当作起点,跑一遍 如果是1为起点的话就把 d[1] 到 d[n]累加起来, 不是1的话 就累加a[1]

然后自信提交 , 果不其然 喜提4个tle

然后想一想,有太多的操作是没用的,太浪费时间了,然后我想了想看能不能反向建边

这样就不用跑那么多次了, 就有了下面的代码

ac代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 1e6+9;
struct edge{ //边
    ll x,w; //x出点  w权值
    bool operator  < (const edge & v)const //重载,得到小根堆
    {
        return w==v.w?x<v.x:w>v.w;
    }
};
int n,m;
ll d[N],sum; // 求和
vector<edge>g[N]; //存边
vector<edge>f[N]; //存反边
void dijkstra(int st){
    memset(d,0x3f,sizeof d); 
    bitset<N>vis; //判断重复
    priority_queue<edge>pq;
    d[st]=0; // 起始为0
    pq.push({st,d[st]});
    while(pq.size()){
        ll x = pq.top().x;pq.pop();
        if(vis[x])continue;//重复就跳过
        vis[x]=1;
        for(auto &[y,w]:g[x]){
            if(!vis[y]&&d[y]>d[x]+w) //更新
            {
                d[y]=d[x]+w;
                pq.push({y,d[y]});
            }
        }
    }
    for(int i=1;i<=n;i++) //求和
        sum += d[i];     
}
void fdijkstra(int st){ //这个和上面dijkstra基本一样,只有读边的时候,换成f[n]
    memset(d,0x3f,sizeof d);
    bitset<N>vis;
    priority_queue<edge>pq;
    d[st]=0;
    pq.push({st,d[st]});
    while(pq.size()){
        ll x = pq.top().x;pq.pop();
        if(vis[x])continue;
        vis[x]=1;
        for(auto &[y,w]:f[x]){
            if(!vis[y]&&d[y]>d[x]+w)
            {
                d[y]=d[x]+w;
                pq.push({y,d[y]});
            }
        }
    }
    for(int i=1;i<=n;i++) sum+=d[i];
}
void solve(){
    
    cin >> n >> m;
    for(int i=1;i<=m;i++){
        int u,v,w;cin >>u >> v >> w;
        if(u!=v) g[u].push_back({v,w}); //存边
        if(u!=v) f[v].push_back({u,w}); //反
    }
    dijkstra(1); //调用
    fdijkstra(1);
    cout << sum; 

}
int main(){
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    int _ =1;
    //cin >> _;
    while(_--)solve();
}