Layout——最短路+差分约束

一共有n头牛，有ml个关系好的牛的信息，有md个关系不好的牛的信息，对应输入的第一行的三个元素，接下来ml行，每行三个元素A,B,D，表示A牛和B牛相距不希望超过D，接下来md行，每行三个元素A,B,D表示A牛和B牛的相距至少要有D才行。求1号牛和n号牛的最大距离，如果距离无限大输出-2，如果无解输出-1。
显然是一道差分约束题
差分约束格式
将原不等式化成u-v<=w的格式

u-v<=w ⇒ add(u,v,w);
u-v>=w ⇒ v-u<=w ⇒ add(v,u,-w);
u-v==0 ⇒ v-u<=0，u-v<=0 ⇒ add(v,u,0),add(u,v,0);
然后跑个最短路就完事了
code:

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;
#define maxn 1010
int head[maxn],dis[1010],vis[1010],num[1010],flag,cnt;
int n,ml,md;
struct node{
	int to;
	int from;
	int w;
}edge[maxn*maxn];
void add(int u,int v,int w){
	cnt++;
	edge[cnt].w=w;
	edge[cnt].to=v;
	edge[cnt].from=head[u];
	head[u]=cnt;
}
void spfa(int s){
	memset(dis,0x3f,sizeof dis);
	vis[s] = 1;
	queue<int>q;
	q.push(s);
	dis[s] = 0;
	num[s]=1;
	while(!q.empty())
	{
		int x = q.front();
		q.pop();
		vis[x] = 0;
		for(int i=head[x];i;i=edge[i].from)
		{
			int to = edge[i].to;
			if(dis[to]>dis[x]+edge[i].w)
			{
				dis[to] = dis[x]+edge[i].w;
				if(!vis[to]){
					vis[to] = 1;
					q.push(to);
					num[to]++;
				}
			}
			if(num[to]>n){
				flag = 1;
				return;
			}
		}
	}
}
int main()
{
	cin>>n>>ml>>md;
	for(int i=1;i<=ml;i++)
	{
		int a,b,d;
		scanf("%d%d%d",&a,&b,&d);
		add(a,b,d);
	}
	for(int i=1;i<=md;i++)
	{
		int a,b,d;
		scanf("%d%d%d",&a,&b,&d);
		add(b,a,-d);
	}
	spfa(1);
	if(flag){
		printf("-1\n");
		return 0;
	}
	if(dis[n]==0x3f3f3f3f){
		printf("-2\n");
		return 0;
	}
	else printf("%d\n",dis[n]);
}