题意
给一个有向图,初始节点为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;
}