【题目来源】:https://vjudge.net/problem/POJ-1511
【题意】
一大批志愿者去一大批的公交站点发传单,这里的一大批数量为n, (^ ,^),完了之后呢,还有回到出发点,题中描述出发点为编号1的站点,然后,问题就变成了这样的一个模型:
求从1到其余个点需要的最短路径之和,加上,其余各点到1号点的最短路径之和。
【思路】
两次spfa。但是由于站点的数量较多,可以用邻接表存图,给出邻接表博客:啊哈雷。
【代码】
#include<cmath>
#include<cstdio>
#include<queue>
#include<vector>
#include<cstring>
#include<algorithm>
#define INF 0x3f3f3f3f
using namespace std;
int d[1000010];
bool vis[1000010];
int first[1000000+10];
int nexx[1000000+10];
int n,m;
long long sum;
struct pp
{
int u,v,x;
} edge[1000000+10];
struct cmp
{
bool operator() (int &a,int &b)
{
return a>b;
}
};
void init()
{
for(int i=1; i<=n; i++)//初始化
first[i]=nexx[i]=-1;
for(int i=1; i<=m; i++)
{
scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].x);
nexx[i]=first[edge[i].u];//重点
first[edge[i].u]=i;
}
}
void spfa(bool flag)
{
for(int i=1; i<=n; i++)
{
d[i]=i==1?0:INF;
vis[i]=0;
}
priority_queue<int,vector<int>,cmp>que;
que.push(1);
vis[1]=1;
while(!que.empty())
{
int res=que.top();
que.pop();
vis[res]=0;
for(int i=first[res]; i!=-1; i=nexx[i])//邻接表的使用
{
int now=(flag?edge[i].v:edge[i].u);//为了判断,用左端点还是右端点,毕竟交换之后有所变化
if(d[now]>d[res]+edge[i].x)
{
d[now]=d[res]+edge[i].x;
if(!vis[now])
{
que.push(now);
vis[now]=1;
}
}
}
}
for(int i=1; i<=n; i++)
sum+=d[i];
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&m);
init();
sum=0;
spfa(1);
for(int i=1; i<=n; i++)
{
first[i]=nexx[i]=-1;
}
for(int i=1; i<=m; i++)
{
nexx[i]=first[edge[i].v];
first[edge[i].v]=i;
}
spfa(0);
printf("%lld\n",sum);
}
}