【差分约束】P1993 小 K 的农场

建立一个超级源点s，使得dis[x]>dis[i]

然后就可以建边 ，a=b 可以拆成b-a0 和 a-b0

之后再跑spfa，判断是否有负环，有负环则说明不能满足，无负环则能满足

 

代码

#include<bits/stdc++.h>
using namespace std;
const int maxm=5005;
const int inf=0x3f3f3f3f;
struct edge
{
	int to,v,nxt;
}G[maxm*3];
int head[maxm],update[maxm],vis[maxm],dis[maxm];
int n,m,cnt;
void add(int x ,int y ,int z)
{
	G[++cnt].to=y; G[cnt].nxt=head[x]; head[x]=cnt; G[cnt].v=z;
}
bool spfa(int x)
{
	queue <int> q;
	q.push(x); dis[x]=0; vis[x]=1;
	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]++;
				if(update[to]==n) return false;
				if(!vis[to])
				{
					vis[to]=1;
					q.push(to);
				}
			}
		}
	}
	return true;
}
int main()
{
	freopen("a.in","r",stdin);
	int op,x,y,z;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;i++)
	{
		scanf("%d",&op);
		if(op==1) scanf("%d%d%d",&x,&y,&z),add(x,y,-z);
		if(op==2) scanf("%d%d%d",&x,&y,&z),add(y,x,z);
		if(op==3) scanf("%d%d",&x,&y),add(x,y,0),add(x,y,0);
	}	
	for(int i=1;i<=n;i++)
		add(n+1,i,0);
	memset(dis,inf,sizeof(dis));
	if(spfa(n+1)==true) printf("Yes");
	else printf("No");
	return 0;
}