洛谷P1993 小K的农场

题目描述
小K在MC里面建立很多很多的农场，总共n个，以至于他自己都忘记了每个农场中种植作物的具体数量了，他只记得一些含糊的信息（共m个），以下列三种形式描述：

农场a比农场b至少多种植了c个单位的作物，
农场a比农场b至多多种植了c个单位的作物，
农场a与农场b种植的作物数一样多。
但是，由于小K的记忆有些偏差，所以他想要知道存不存在一种情况，使得农场的种植作物数量与他记忆中的所有信息吻合。

输入输出格式
输入格式：
第一行包括两个整数 n 和 m，分别表示农场数目和小 K 记忆中的信息数目。

接下来 m 行：

如果每行的第一个数是 1，接下来有 3 个整数 a,b,c，表示农场 a 比农场 b 至少多种植了 c 个单位的作物。

如果每行的第一个数是 2，接下来有 3 个整数 a,b,c，表示农场 a 比农场 b 至多多种植了 c 个单位的作物。如果每行的第一个数是 3，接下来有 2 个整数 a,b，表示农场 a 种植的的数量和 b 一样多。

输出格式：
如果存在某种情况与小 K 的记忆吻合，输出“Yes”，否则输出“No”。

输入输出样例
输入样例#1： 
3 3
3 1 2
1 1 3 1
2 2 3 2
输出样例#1： 
Yes
说明
对于 100% 的数据保证：1 ≤ n，m，a，b，c ≤ 10000。

代码

// luogu-judger-enable-o2
#include <bits/stdc++.h>
using namespace std;
struct node{
	int v,w;
	node (){ }
	node (int _v,int _w){
		v=_v;
		w=_w;
	}
};

queue<int>qu;

int inf=0x3f3f3f3f;

vector<node> g[10010];
int inq[10010],dst[10010];
int n,m,cnt[10010];

void add(int u,int v,int w){
	g[u].push_back(node(v,w));
	g[v].push_back(node(u,w));
}

int spfa(int s){
	memset(dst,inf,sizeof dst);
	int u=s;
	dst[u]=0;
	inq[u]=1;
	qu.push(u);
	cnt[u]++;
	while (!qu.empty()){
		u=qu.front();
		qu.pop();
		inq[u]=0;
		for (int i=0;i<g[u].size();i++){
			int v=g[u][i].v;
			int w=g[u][i].w;
			if (dst[v]>dst[u]+w){
				dst[v]=dst[u]+w;
				if (!inq[v]){
					qu.push(v);
					inq[v]=1;
					cnt[v]++;
					if (cnt[v]>n){
						return -1;
					}
				}
			}
		}
	}
	return 1;
}


int main(){
	cin>>n>>m;
	
	
	for (int i=0;i<n;i++){
		int a,b,c;
		cin>>a>>b>>c;
		if (a==1)
		{
			int d;
			cin>>d;
			g[b].push_back(node(c,-d));
			g[0].push_back(node(c,-d));
		}
		else if(a==2)
		{
			int d;
			cin>>d;
			g[c].push_back(node(b,d));
			g[0].push_back(node(b,d));
		}
		else if(a==3)
		{
			g[b].push_back(node(c,0));
			g[c].push_back(node(b,0));
			g[0].push_back(node(c,0));
			g[0].push_back(node(b,0));
		}
	}
	
	if (spfa(0)==-1)
	{
		cout<<"No";
	}
	else 
	{
		cout<<"Yes";
	}
	return 0;
}

温馨提示：记得开O2