POJ 1511 链式前向星+SPFA

#include<iostream>
#include<cstdio>
#include<cstdlib>
using namespace std;
const int maxn=1000010,inf=1000000000;
long long ans;
int e,to[maxn],next[maxn],begin[maxn],w[maxn];
int e1,to1[maxn],next1[maxn],begin1[maxn],w1[maxn];
int d[maxn],p[maxn],q[maxn*50];
int m,n,f,l;
void add(int x,int y,int z){
to[++e]=y;
next[e]=begin[x];
begin[x]=e;
w[e]=z;
}
void add1(int x,int y,int z){
to1[++e1]=y;
next1[e1]=begin1[x];
begin1[x]=e1;
w1[e1]=z;
}
int main(){
int i,j,k,m,n;
int t;
scanf("%d",&t);
while(t--){
int i,j,k,x,y,z;
scanf("%d%d",&n,&m);
e=0;e1=0;
for(i=1;i<=n;i++){begin[i]=0;begin1[i]=0;}
for(i=1;i<=m;i++){
scanf("%d%d%d",&x,&y,&z);
add(x,y,z);
add1(y,x,z);
}
for(i=1;i<=n;i++){
d[i]=inf;
p[i]=0;
}
f=0;l=1;
d[1]=0;q[1]=1;p[1]=1;
while(f<l){
f++;
k=q[f];p[k]=0;
for(i=begin[k];i;i=next[i])
if(d[to[i]]>d[k]+w[i]){
d[to[i]]=d[k]+w[i];
if(!p[to[i]]){
q[++l]=to[i];
p[to[i]]=1;
}
}
}
ans=0;
for(i=1;i<=n;i++)ans+=d[i];

for(i=1;i<=n;i++){
d[i]=inf;
p[i]=0;
}
f=0;l=1;
d[1]=0;q[1]=1;p[1]=1;
while(f<l){
f++;
k=q[f];p[k]=0;
for(i=begin1[k];i;i=next1[i])
if(d[to1[i]]>d[k]+w1[i]){
d[to1[i]]=d[k]+w1[i];
if(!p[to1[i]]){
q[++l]=to1[i];
p[to1[i]]=1;
}
}
}
for(i=1;i<=n;i++)ans+=d[i];
printf("%I64d\n",ans);
}
return 0;
}