POJ 3169 差分约束系统

大致题意：

    有n头牛，按照序号1……n排队，多头牛可以站在同一个位置(暂且这么认为)，也就是任意两头牛之间的距离都大于等于0。先给出ml组约束关系(u,v,w)代表第u牛和第v牛之间的距离要小于等于w。再给出md组关系(u,v,w)，代表第u牛和第v牛之间的距离要大于等于w。求这n头牛排成的队伍能否符合以上的约束，不能的话(也就是出现负环)，输出“-1”，如果距离是inf，输出“-2”。否则输出这个最短距离。

大致思路：

    转化为差分约束模型。dis[i]表示第i头牛所在的位置。如果spfa判断出负环的话就输出-1，如果dis[n]==inf则输出-2。否则输出dis[n]。

#include<cstdio>
#include<queue>
#include<cstring>
using namespace std;

const int inf=1<<30,
		maxv=1005,
		maxe=20005;
struct edge
{
	int u,v,k,next;
	void init(int tu,int tv,int tk,int tnext)
	{
		u=tu, v=tv, k=tk, next=tnext;
	}
}e[maxe];
int link[maxv],cnt[maxv],d[maxv];
int ne;
bool vis[maxv];

void addedge(int u,int v,int k)
{
	e[ne].init(u,v,k,link[u]);
	link[u]=ne++;
}

int spfa(int s,int t)
{
	int i,n=t-s;
	for(i=s;i<=t;i++)
	{
		d[i]=inf;
	}
	queue<int> q;
	memset(vis,0,sizeof(vis));
	memset(cnt,0,sizeof(cnt));
	q.push(s);
	vis[s]=true;
	d[s]=0;
	cnt[s]++;
	while(!q.empty())
	{
		int p,v,u=q.front();
		q.pop();
		vis[u]=false;
		for(p=link[u];p!=-1;p=e[p].next)
		{
			v=e[p].v;
			if(d[v]>d[u]+e[p].k)
			{
				d[v]=d[u]+e[p].k;
				cnt[v]++;
				if(cnt[v]>n)
				{
					return -1;
				}
				if(!vis[v])
				{
					vis[v]=true;
					q.push(v);
				}
			}
		}
	}
	if(d[t]==inf)
	{
		return -2;
	}
	return d[t];
}

int main()
{
	int n,ml,md;
	while(~scanf("%d%d%d",&n,&ml,&md))
	{
		ne=0;
		memset(link,-1,sizeof(link));
		int u,v,k;
		while(ml--)
		{
			scanf("%d%d%d",&u,&v,&k);
			addedge(u,v,k);
		}
		while(md--)
		{
			scanf("%d%d%d",&u,&v,&k);
			addedge(v,u,-k);
		}
		printf("%d\n",spfa(1,n));
	}
	return 0;
}