poj1511

题意：

给定节点数n,和边数m,边是单向边.

问从1节点出发到2,3,...n 这些节点路程和从从这些节点回来到节点1的路程和最小值。

思路：

由于点和边都有1e6大小，还有多组数据，直接用普通的dijkstra算法自然是不行的。于是用SPFA算法，复杂度为O(ke),刚好能够卡过。

#include<cstdio>
#include<cstring>
#include<queue>
#define N 1000009
#define inf 1000000005
using namespace std;
int cnt,cnt2,p,q;
queue<int> que;
int vis[N],dis[N];
struct node
{
	int v,w,next;
}s[N],s2[N];
int head[N];
int head2[N];
void add(int u,int v,int w)
{
	s[cnt].v=v;
	s[cnt].w=w;
	s[cnt].next=head[u];
	head[u]=cnt++;
	
	s2[cnt2].v=u;
	s2[cnt2].w=w;
	s2[cnt2].next=head2[v];
	head2[v]=cnt2++;
}
void spfa(int src)
{
	while(!que.empty())	que.pop(); 
	int v;
	memset(vis,0,sizeof(vis));
	for(int i=1;i<=p;i++)
	dis[i]=inf;
	dis[src]=0;
	vis[src]=1;
	que.push(src);
	while(!que.empty())
	{
		int pre=que.front();
		que.pop();
		vis[pre]=0;
		for(int i=head[pre];i!=-1;i=s[i].next)
		{
			v=s[i].v;
			if(dis[pre]+s[i].w<dis[v])
			{
				dis[v]=dis[pre]+s[i].w;
				if(!vis[v])
				{
					que.push(v);
					vis[v]=1;
				}
			}
		}
	}
}
void spfa2(int src)
{
	while(!que.empty())	que.pop(); 
	int v;
	memset(vis,0,sizeof(vis));
	for(int i=1;i<=p;i++)
	dis[i]=inf;
	dis[src]=0;
	vis[src]=1;
	que.push(src);
	while(!que.empty())
	{
		int pre=que.front();
		que.pop();
		vis[pre]=0;
		for(int i=head2[pre];i!=-1;i=s2[i].next)
		{
			v=s2[i].v;
			if(dis[pre]+s2[i].w<dis[v])
			{
				dis[v]=dis[pre]+s2[i].w;
				if(!vis[v])
				{
					que.push(v);
					vis[v]=1;
				}
			}
		}
	}
}
int main()
{
	//FILE *fp=fopen("t.txt","r");
	long long ans;
	int u,v,w,ca;
	scanf("%d",&ca);
	while(ca--)
	{
		scanf("%d%d",&p,&q);
		cnt=0;cnt2=0;
		memset(head,-1,sizeof(head));
		memset(head2,-1,sizeof(head2));
		for(int i=0;i<q;i++)
		{
			scanf("%d%d%d",&u,&v,&w);
			add(u,v,w);
		}
		ans=0;
		spfa(1);
		for(int i=2;i<=p;i++)
		ans+=dis[i];
		spfa2(1);
		for(int i=2;i<=p;i++)
		ans+=dis[i];
		printf("%lld\n",ans);
	}
	return 0;
}