河南萌新联赛2024第（五）场 : F-小美想跑步

题意

给一个有向图，初始节点为1，每次访问完一个节点后必须找到一条返回1节点的路径，求访问完所有结点所需要的最小总边权

思路

先求1结点到其他节点的最短路径，即单源最短路径。

然后由其他节点返回1结点的最短路径，我们可以反向建图，将原来的有向图方向反转，再依然按照结点1的单源最短路径问题，就得到了有其他节点返回1节点的最短路径

代码

/*I lv ya*/
#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <iostream>
#include <algorithm>
#include <vector>
#include <stack>
#include <sstream>
#include <map>
#include <set>
#include <queue>
#include <stdlib.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
const int N = 2e6+5;
int n,m;
int head1[N],cnt1=0;
int head2[N],cnt2=0;
int dis1[N],dis2[N];
bool vis[N];
struct edge
{
    int t,ne,w;
}ed1[N],ed2[N];
void add1(int s,int t,int w)
{
    ed1[++cnt1].ne=head1[s];
    ed1[cnt1].w=w;
    ed1[cnt1].t=t;
    head1[s]=cnt1;
}
void add2(int s,int t,int w)
{
    ed2[++cnt2].ne=head2[s];
    ed2[cnt2].w=w;
    ed2[cnt2].t=t;
    head2[s]=cnt2;
}
void dij1()
{
    memset(dis1,0x3f,sizeof(dis1));
    memset(vis,0,sizeof(vis));
    priority_queue<PII,vector<PII>,greater<PII> >pq;
    dis1[1]=0;
    pq.push({0,1});
    while(!pq.empty())
    {
        PII now=pq.top();
        pq.pop();
        int pos=now.second;
        if(vis[pos]) continue;
        vis[pos]=1;
        for(int i=head1[pos];i;i=ed1[i].ne)
        {
            int to=ed1[i].t;
            if(dis1[to]>dis1[pos]+ed1[i].w)
            {
                dis1[to]=dis1[pos]+ed1[i].w;
                pq.push({dis1[to],to});
            }
        }
    }
}
void dij2()
{
    memset(dis2,0x3f,sizeof(dis2));
    memset(vis,0,sizeof(vis));
    priority_queue<PII,vector<PII>,greater<PII> >pq;
    dis2[1]=0;
    pq.push({0,1});
    while(!pq.empty())
    {
        PII now=pq.top();
        pq.pop();
        int pos=now.second;
        if(vis[pos]) continue;
        vis[pos]=1;
        for(int i=head2[pos];i;i=ed2[i].ne)
        {
            int to=ed2[i].t;
            if(dis2[to]>dis2[pos]+ed2[i].w)
            {
                dis2[to]=dis2[pos]+ed2[i].w;
                pq.push({dis2[to],to});
            }
        }
    }
}
void solve()
{
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        int s,t,w;
        cin>>s>>t>>w;
        add1(s,t,w);
        add2(t,s,w);
    }
    dij1();dij2();
    ll ans=0;
    for(int i=2;i<=n;i++)
    {
        ans+=dis1[i]+dis2[i];
    }
    cout<<ans<<endl;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(NULL),cout.tie(NULL);
    int _;
    _=1;
    while(_--)
    {
        solve();
    }
    return 0;
}