hdu 1535 Invitation Cards（SPFA）

hdu 1535
Invitation Cards

题意：在一个有向图中，一个人需要走完所有景点，然后返回，求起点到所有景点的最短路+所有景点到起点的最短路。

解题思路：正向求一次SPFA，然后将所有边反向，再求一次SPFA。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <stack>
#include <cmath>
#include <vector>
#include <queue>
#include <numeric>
#include <map>
using namespace std;
#define LL long long
#define M 1002
#define N 1000002
#define INF 1<<29
#define CLS(x,v) memset(x,v,sizeof(x))
#define FOR(i,a,n) for(int i=(a);i<=(n);++i)
//建立反向边，求两遍SPFA
//注意一开始 在函数里面开 10^6 大的数组，会堆栈溢出

int headlist1
;
int headlist2
;
struct EDGE{
int to,w,next;
}e1
,e2
;
int n;
bool vis
;
int dist
;
int spfa(int s,int head[],EDGE e[])
{
queue<int> Q;
Q.push(s);
FOR(i,0,N)dist[i]=INF;
CLS(vis,0);
dist[s]=0;
int pre,t;
while(!Q.empty())
{
pre=Q.front();
Q.pop();
vis[pre]=0;
for(int i=head[pre];i!=-1;i=e[i].next)
{
t=e[i].to;
if(dist[t]>dist[pre]+e[i].w)
{
dist[t]=dist[pre]+e[i].w;
if(!vis[t])
{
Q.push(t);
vis[t]=1;
}
}
}
}
int ret=0;
for(int i=1;i<=n;i++)
ret=ret+dist[i];
return ret;
}
int main()
{
int T,m;
cin>>T;
int s,t,w;
while(T--)
{
scanf("%d%d",&n,&m);
CLS(headlist1,-1);
CLS(headlist2,-1);
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&s,&t,&w);
e1[i].to=t;
e1[i].w=w;
e1[i].next=headlist1[s];
headlist1[s]=i;//记录边的编号
//建立反向边
e2[i].to=s;
e2[i].w=w;
e2[i].next=headlist2[t];
headlist2[t]=i;
}
printf("%d\n",spfa(1,headlist1,e1)+spfa(1,headlist2,e2));
}

return 0;
}