POJ 1511 Invitation Cards（spfa最短路）

大概题意：有n个点需要志愿者去发传单，然后有m条公交车路线..单向的，早上这些志愿去除了大本营1之外的所有点发传单，然后晚上再回来...问全部志愿者一来一回的最短路程是多少...（大概一个点只有一个志愿者吧...）

类似于POJ 3268 那个牛聚会的那个题目...从1点出发在回来1点...那么1点出发到各点的最短路可以spfa一下...然后从各点回到1点可以反向建边，然后以1为起点spfa计算一下单源最短路径就好了...两个加起来的和就是答案了

第一次用邻接表...找了学长的邻接表模板...貌似和我百度的不一样..用着还顺手...

#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<queue>
#include<algorithm>
#include<iostream>
using namespace std;
#define M 1000009
#define inf 0x7fffffff
struct node
{
int u,v,w,next;
}edge1[M+M],edge2[M+M];
int head1[M],head2[M];
int n,m,t1,t2,used[M],dis[M];
void inti()
{
t1=t2=0;
memset(head1,-1,sizeof(head1));
memset(head2,-1,sizeof(head2));
}
void add(int u,int v,int w)
{
edge1[t1].v=v;
edge1[t1].w=w;
edge1[t1].next=head1[u];
head1[u]=t1++;

edge2[t2].v=u;
edge2[t2].w=w;
edge2[t2].next=head2[v];
head2[v]=t2++;
}
long long spfa(node edge[],int head[])
{
for(int i=1;i<=n;i++)
dis[i]=inf;
queue<int> q;
memset(used,0,sizeof(used));
q.push(1);
dis[1]=0;
used[1]=1;
while(!q.empty())
{
int u=q.front();
q.pop();
used[u]=0;
for(int i=head[u];i!=-1;i=edge[i].next)
{
int v=edge[i].v;
if(dis[v]>edge[i].w+dis[u])
{
dis[v]=edge[i].w+dis[u];
if(used[v]==0)
{
used[v]=1;
q.push(v);
}
}
}
}
long long ans=0;
for(int i=1;i<=n;i++)
ans+=dis[i];
return ans;
}
int main()
{
int T;
cin>>T;
while(T--)
{
int u,v,w;
scanf("%d %d",&n,&m);
inti();
for(int i=0;i<m;i++)
{
scanf("%d %d %d",&u,&v,&w);
add(u,v,w);
}
long long ans;
ans=spfa(edge1,head1)+spfa(edge2,head2);
cout<<ans<<endl;
}
return 0;
}