SPFA最短路 POJ1511 Invitation Card

題目大意：有一個n個頂點，m 條有向邊的圖，求一個頂點到p個頂點並且回來的最短距離。

分析：使用spfa算法。鄰接表 存儲邊。前向星時間爆了。循環隊列竟然導致了超時。注意，本題的空間充足。

#include <iostream>
#include <cstring>
#include <cstdio>
#define MAXN 1000000
#define MAXM 1000000
#define INF 1100000000
using namespace std;

int u[MAXN+5],v[MAXN+5];
long long w[MAXN+5];
int q[10000000];
long long f[MAXN+10];
bool vis[MAXN+10];
int n,m;

struct Edge {
	int to, next;
	long long len;
}edge[MAXM +10];
int en, head[MAXN];
inline void add_edge(int u, int v, long long len) {
	edge[en].to = v, edge[en].len = len, edge[en].next = head[u], head[u] = en++;
}

int spfa()
{
	int i,h,t;
	memset(vis,false,sizeof(vis));
	for (i=1;i<=n;i++) f[i]=INF;
	h=1;t=1;q[1]=1;vis[1]=true;f[1]=0;
	while (h!=t+1)
	{
		i=q[h];vis[i]=false;
		for (int p=head[i];p!=-1;p=edge[p].next)
		{
		    int to = edge[p].to, len = edge[p].len;
			if (f[to]>f[i]+len)
			{
				f[to]=f[i]+len;
				if (!vis[to])
				{
					t++;
					q[t]=to;
					vis[to]=true;
				}
			}
		}
		h++;
	}
	return 0;
}

int main()
{
	int cases,x,y;
	long long z,sum;
	scanf("%d",&cases);
	while (cases--)
	{
		scanf("%d%d",&n,&m);
		for (int i=1;i<=m;i++)
		{
			scanf("%d%d%lld",&x,&y,&z);
			u[i]=x;v[i]=y;w[i]=z;
		}
		fill(head + 1, head + 1 + n, -1);
		en=0;
		for (int i=1;i<=m;i++) add_edge(u[i],v[i],w[i]);
		sum=0;
		spfa();
		for (int i=1;i<=n;i++) sum+=f[i];
		
		fill(head + 1, head + 1 + n, -1);
		en=0;
		for (int i=1;i<=m;i++) add_edge(v[i],u[i],w[i]);
		spfa();
		for (int i=1;i<=n;i++) sum+=f[i];
		printf("%lld\n",sum);
	}
	return 0;
}