在求解最短路时,单对多最短路径,多对多都已经有现成的算法,那么怎么求多对单?
很简单,只需要反向建图再用spfa,ford,dijkstra等算法即可,下面例题正好就是单对多和多对单的应用。
题目链接:点击这里
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<string>
#include<queue>
#define maxd 1000+5
#define maxm 100000+5
using namespace std;
struct edge{
int to,next,w;
}kkk[maxm],fedge[maxm];
int head[maxd],dis[maxd],cnt=0,fcnt=0,fhead[maxd],fdis[maxd];
bool visit[maxd],fvisit[maxd];
void addedge(int u,int v,int w){
cnt++;
kkk[cnt].to=v;
kkk[cnt].w=w;
kkk[cnt].next=head[u];
head[u]=cnt;
fedge[cnt].to=u;
fedge[cnt].w=w;
fedge[cnt].next=fhead[v];
fhead[v]=cnt;
}//链式前向星存边
struct node{
int dis;//权重
int pos;//位置
bool operator <( const node &x )const{
return x.dis < dis;
}
};
priority_queue<node>q;//优先队列优化
priority_queue<node>fq;
int u,v,w;
int main(){
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++) dis[i]=fdis[i]=2147483647;//dis初始化
memset(visit,0,sizeof(visit));
memset(fvisit,0,sizeof(fvisit));
for(int i=1;i<=m;i++){
scanf("%d%d%d",&u,&v,&w);
addedge(u,v,w);
}//读边
dis[1]=0;
q.push((node){0,1});//入队
while(!q.empty()){
node p=q.top();
q.pop();
int x=p.pos;
if(visit[x]) continue;
visit[x]=1;
for(int i=head[x];i;i=kkk[i].next){
if(dis[kkk[i].to]>dis[x]+kkk[i].w){
dis[kkk[i].to]=dis[x]+kkk[i].w;
if( !visit[kkk[i].to] ){
q.push( ( node ){dis[kkk[i].to], kkk[i].to} );
}
}
}
}
int ans=0;
fdis[1]=0;
fq.push((node){0,1});//入队
while(!fq.empty()){
node p=fq.top();
fq.pop();
int x=p.pos;
if(fvisit[x]) continue;
fvisit[x]=1;
for(int i=fhead[x];i;i=fedge[i].next){
if(fdis[fedge[i].to]>fdis[x]+fedge[i].w){
fdis[fedge[i].to]=fdis[x]+fedge[i].w;
if( !fvisit[fedge[i].to] ){
fq.push( ( node ){fdis[fedge[i].to],fedge[i].to} );
}
}
}
}
for(int i=2;i<=n;i++) {
ans=ans+dis[i]+fdis[i];
}
cout<<ans;
return 0;
}