HDU 1535 Invitation Cards(SPFA+双向建图 )

题目大意:求一个从地点然后再从一个地点返回过来,的最短路径.

思路:正向+逆向建图

#include<iostream>
#include<cstring>
#include<queue>
#include<cstdio>
#define inf 0x3f3f3f3f
using namespace std;
bool vis[1000010];
int n,head1[1000010],s1,head2[1000010],s2,ans,dis[1000010],dis2[1000010];

struct node
{
int to,w,next;
}q1[1000010],q2[1000010];

void add1(int a,int b,int c)
{
q1[s1].to=b;
q1[s1].w=c;
q1[s1].next=head1[a];
head1[a]=s1++;
}
void add2(int a,int b,int c)
{
q2[s2].to=a;
q2[s2].w=c;
q2[s2].next=head2[b];
head2[b]=s2++;
}
void SPFA1(int st)
{
int k;
queue<int>Q;
for(int i=1;i<=n;i++)
dis[i]=inf;

memset(vis,false,sizeof(vis));

dis[st]=0;
vis[st]=true;
Q.push(st);
while(!Q.empty())
{
k=Q.front();
Q.pop();
vis[k]=false;
for(int i=head1[k];i!=-1;i=q1[i].next)
{
int v=q1[i].to;
if(dis[v]>dis[k]+q1[i].w)
{
dis[v]=dis[k]+q1[i].w;
if(!vis[v])
{
vis[v]=true;
Q.push(v);
}
}
}
}
}
void SPFA2(int st)
{
queue<int >Q;
int i,j,k;
memset(vis,false,sizeof(vis));
for(i=1;i<=n;i++)
{
dis2[i]=inf;
}
dis2[st]=0;
vis[st]=true;
Q.push(st);
while(!Q.empty())
{
k=Q.front();
Q.pop();
vis[k]=false;
for(i=head2[k];i!=-1;i=q2[i].next)
{
int u=q2[i].to;
if(dis2[u]>dis2[k ]+q2[i].w)
{
dis2[u]=dis2[k]+q2[i].w;
if(!vis[u])
{
vis[u]=true;
Q.push(u);
}
}
}
}
}
int main()
{
int m,i,j,t,a,b,c;
int cla;
scanf("%d",&cla);
while(cla--)
{
ans=0;
s2=s1=1;
memset(head1,-1,sizeof(head1));
memset(head2,-1,sizeof(head2));
scanf("%d%d",&n,&m);
for(i=0;i<m;i++)
{
scanf("%d%d%d",&a,&b,&c);
add1(a,b,c);
add2(a,b,c);
}
SPFA1(1);
SPFA2(1);
for(i=1;i<=n;i++)
{
ans+=dis[i]+dis2[i];
}
printf("%d\n",ans);
}
return 0;
}