题目
题解
这道题目跟模板差不太多,不同的是它还要求每个点到起点的最短距离,因为我才刚开始学 , 第一时间我想到就是把每个点都当作起点,跑一遍 如果是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();
}