【差分约束】P4878 [USACO05DEC]Layout G

差分约束建边后，两次spfa，一次判断有无负环，另一次用于计算答案

 

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=1005;
const int maxm=50005;
const int inf=0x3f3f3f3f;
int n,ml,cnt,mr,update[maxn],head[maxn],dis[maxn],vis[maxn];
struct edge
{
	int to,nxt,v;
}G[maxm];
void add(int x,int y,int z)
{
	G[++cnt].to=y; G[cnt].nxt=head[x];G[cnt].v=z; head[x]=cnt;
}
void spfa(int s)
{
	queue <int> q;
	memset(dis,0x3f,sizeof(dis)); 
	for(int i=0;i<=n;i++) vis[i]=0,update[i]=0;
	q.push(s);
	vis[s]=1; dis[s]=0;
	while(!q.empty())
	{
		int u=q.front();
		q.pop(); vis[u]=0;
		for(int i=head[u];i;i=G[i].nxt)
		{
			int to=G[i].to;
			int v=G[i].v;
			if(dis[to]>dis[u]+v)
			{
				dis[to]=dis[u]+v;
				update[to]=update[u]+1;
				if(update[to]>=n)
				{
					printf("-1");
					exit (0);
				}
				if(!vis[to])
				{
					q.push(to);
					vis[to]=1;
				}
			}
		}
	}
}
int main()
{
	freopen("layout.2.in","r",stdin);
	scanf("%d%d%d",&n,&ml,&mr);
	int x,y,z;
	for(int i=1;i<=ml;i++)
	{
		scanf("%d%d%d",&x,&y,&z);
		add(x,y,z);
	}
	for(int i=1;i<=mr;i++)
	{
		scanf("%d%d%d",&x,&y,&z);
		add(y,x,-z);
	}
	for(int i=1;i<=n;i++) add(0,i,0);
	spfa(0); //判负环
	spfa(1); //求结果
	if(dis[n]==inf) printf("-2");
	else printf("%d",dis[n]); 
	return 0;
}