来回最短路

F-小美想跑步_河南萌新联赛2024第（五）场：信息工程大学 (nowcoder.com)

#include "bits/stdc++.h"
#define int long long
#define pii pair<int,int>
using namespace std;
int e[1000000],ne[1000000],h[1000000],idx;
int dist[10000000],st[10000000],w[1000000];
int u[10000000],v[10000000],d[10000000];
int n,m;
void add(int x,int y,int z)
{
	e[idx]=y,w[idx]=z,ne[idx]=h[x],h[x]=idx++;
}
void bfs(int x)
{
    priority_queue<pii,vector<pii>,greater<pii>>q;
    dist[x]=0;
    q.push({0,x});
    while(q.size()!=0)
    {
        auto it=q.top();
        q.pop();
        int ver=it.second;
        if(st[ver]!=0)
                continue;
         st[ver]=1;
      	for(int i=h[ver];i!=-1;i=ne[i])
	{
	          int j=e[i];
                if(dist[j]>dist[ver]+w[i])
                {
                    dist[j]=dist[ver]+w[i];
                    q.push({dist[j],j});
                }
	}  
    }
}
signed main()
{
	ios::sync_with_stdio(0);
	cin.tie(0),cout.tie(0);
    cin>>n>>m;
    int sum=0;
    memset(h,-1,sizeof(h));
   for(int i=1;i<=m;i++)
   {
       cin>>u[i]>>v[i]>>d[i];
       add(u[i],v[i],d[i]);
   }
 memset(dist,0x3f,sizeof(dist));
 bfs(1);
    for(int i=1;i<=n;i++)
    sum+=dist[i];
     memset(h,-1,sizeof(h));
      memset(dist,0x3f,sizeof(dist));
       memset(st,0,sizeof(st));
       for(int i=1;i<=m;i++)
   {
       add(v[i],u[i],d[i]);
   }
    bfs(1);
    for(int i=1;i<=n;i++)
    sum+=dist[i];
    cout<<sum;
 }