hdu1535 SPFA

#include<stdio.h>
#include<string.h>
#include<queue>
#define INF 1000000000
#define ii __int64
using namespace std;
struct node
{
ii v;
ii val;
ii next;
}edge1[1000003],edge2[1000003];
ii head1[1000003],dis1[1000003],head2[1000003],dis2[1000003],vis[1000003];
ii n,m,t,index1,index2;
void add1(ii x,ii y,ii z)
{
ii i,j;
edge1[index1].v=y;
edge1[index1].val=z;
edge1[index1].next=head1[x];
head1[x]=index1++;
}
void add2(ii x,ii y,ii z)
{
ii i,j;
edge2[index2].v=y;
edge2[index2].val=z;
edge2[index2].next=head2[x];
head2[x]=index2++;
}
void SPFA1(ii u)
{
ii i,j;
memset(vis,0,sizeof(vis));
queue<ii>q;
for(i=1;i<=n;i++)
dis1[i]=INF;
dis1[u]=0;
vis[u]=1;
q.push(u);
while(!q.empty())
{
ii v=q.front();
q.pop();
vis[v]=0;
for(i=head1[v];i!=-1;i=edge1[i].next)
{
ii tmp=edge1[i].v;
if(dis1[tmp]>dis1[v]+edge1[i].val)
{
dis1[tmp]=dis1[v]+edge1[i].val;
if(!vis[tmp])
{
q.push(tmp);
vis[tmp]=1;
}
}
}
}
}
void SPFA2(ii u)
{
ii i,j;
memset(vis,0,sizeof(vis));
queue<ii>q;
for(i=1;i<=n;i++)
dis2[i]=INF;
dis2[u]=0;
vis[u]=1;
q.push(u);
while(!q.empty())
{
ii v=q.front();
q.pop();
vis[v]=0;
for(i=head2[v];i!=-1;i=edge2[i].next)
{
ii tmp=edge2[i].v;
if(dis2[tmp]>dis2[v]+edge2[i].val)
{
dis2[tmp]=dis2[v]+edge2[i].val;
if(!vis[tmp])
{
q.push(tmp);
vis[tmp]=1;
}
}
}
}
}
int main()
{
ii i,j,t,m;
scanf("%I64d",&t);
while(t--)
{
scanf("%I64d%I64d",&n,&m);
index1=index2=1;
memset(dis1,0,sizeof(dis1));
memset(head1,-1,sizeof(head1));
memset(dis2,0,sizeof(dis2));
memset(head2,-1,sizeof(head2));
for(i=0;i<m;i++)
{
ii x,y,z;
scanf("%I64d%I64d%I64d",&x,&y,&z);
add1(x,y,z);
add2(y,x,z);
}

SPFA1(1);

/*for(i=1;i<=n;i++)
printf("%d ",dis1[i]);
printf("\n");*/

SPFA2(1);

/*for(i=1;i<=n;i++)
printf("%d ",dis2[i]);printf("\n");*/

ii ans=0;
for(i=1;i<=n;i++)
{
ans+=dis1[i]+dis2[i];
}
printf("%I64d\n",ans);
}
}