poj1511 Invitation Cards SPFA

http://poj.org/problem?id=1511

先一次用SPFA求早上从定点1到各点的最短路径和，再对其求反向图，用SPFA求一次顶点1到各顶点的最短路径和，这个值其实就是晚上各点回到顶点1的最短路径和了。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
typedef long long ll;
const int MAXN=1000005;
const int MAXM=1000005;
const ll INF=0x3f3f3f3f;
int e,vis[MAXN],head[MAXN],next[MAXM],n,m;
ll dis[MAXN];
struct node{
	int s,e;
	ll w;
	node(){}
	node(int s,int e,ll w){
		this->s=s; this->e=e; this->w=w; 
	}
}t[MAXM];
void addnode(int u,int v,ll w){//加边
	t[e]=node(u,v,w);
	next[e]=head[u];
	head[u]=e++;
}
bool relax(int u,int v,ll w){//松弛
	if(dis[v]>dis[u]+w){
		dis[v]=dis[u]+w;
		return true;
	}
	return false;
}
void init(){
	memset(head,-1,sizeof(head));
	memset(next,-1,sizeof(next));
	e=0;
	for(int i=0;i<m;i++){
		int s,e;
		ll w;
		scanf("%d %d %lld",&s,&e,&w);
		addnode(s,e,w);
	}
}
void un_init(){//回去的路
	memset(head,-1,sizeof(head));
	memset(next,-1,sizeof(next));
	e=0;
	for(int i=0;i<m;i++){
		addnode(t[i].e,t[i].s,t[i].w); 
	}	
}
ll get_sum(ll a[]){
	ll res=0;
	for(int i=1;i<=n;i++){
		res+=a[i];
	}
	return res;
}
void spfa(){
	memset(vis,0,sizeof(vis));
	memset(dis,INF,sizeof(dis));
	dis[1]=0; vis[1]=1;
	queue<int>que;
	que.push(1);
	while(que.size()){
		int now=que.front();
		que.pop();
		vis[now]=0;
		for(int i=head[now];i!=-1;i=next[i]){
			if(relax(now,t[i].e,t[i].w)&&!vis[t[i].e]){
				que.push(t[i].e);
				vis[t[i].e]=1;
			}
		}
	}
}
int main(){
	int T;
	scanf("%d",&T);
	while(T--){
		scanf("%d %d",&n,&m);
		init(); spfa();
		ll ans=get_sum(dis); 
		un_init(); spfa();
		ans+=get_sum(dis);
		printf("%lld\n",ans);
	}
	return 0;
}