差分约束。
发现题中每个农场的作物数
x
i
x_i
xi 要满足以下约束关系:
{
x
a
−
x
b
≥
c
1
x
a
−
x
b
≤
c
2
x
a
=
x
b
\begin{cases} x_a-x_b\geq c_1\\ x_a-x_b\leq c_2\\ x_a=x_b \end{cases}
⎩
⎨
⎧xa−xb≥c1xa−xb≤c2xa=xb
把以上关系进行转化,重点是“相等”这一约束关系的转化:
{
x
a
−
x
b
≥
c
1
x
b
−
x
a
≥
−
c
2
x
b
−
x
a
≥
0
x
a
−
x
b
≥
0
\begin{cases} x_a-x_b\geq c_1\\ x_b-x_a\geq -c_2\\ x_b-x_a\geq 0\\ x_a-x_b\geq 0\\ \end{cases}
⎩
⎨
⎧xa−xb≥c1xb−xa≥−c2xb−xa≥0xa−xb≥0
按照上述约束关系,SPFA 求最长路即可。注意此时 dis
要赋值为
−
inf
-\inf
−inf。
注意若图不连通,可能会出现有 x i x_i xi 值为 − inf -\inf −inf 的情况,所以要建立超级源点。
#include <bits/stdc++.h>
using namespace std;
const int maxn=5005,maxm=1e4+5;
int head[maxn],cnt,tot[maxn],dis[maxn],n,m;
bool vis[maxn];
struct edge{int to,nxt,w;}e[maxm];
void add(int x,int y,int z){e[++cnt]={y,head[x],z},head[x]=cnt;}
bool spfa(int s)
{
queue<int> q;
memset(dis,-0x3f3f3f,sizeof dis);
dis[s]=0,q.push(s),vis[s]=1,tot[s]++;
while(!q.empty())
{
int x=q.front();q.pop(),vis[x]=0;
for(int i=head[x];i;i=e[i].nxt)
if(dis[e[i].to]<dis[x]+e[i].w)
{
dis[e[i].to]=dis[x]+e[i].w;
if(!vis[e[i].to]) vis[e[i].to]=1,q.push(e[i].to),tot[e[i].to]++;
if(tot[e[i].to]>=n) return 0;
}
}
return 1;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=m;i++)
{
int op;cin>>op;
if(op==1)
{
int a,b,c;cin>>a>>b>>c;
add(b,a,c);
}
else if(op==2)
{
int a,b,c;cin>>a>>b>>c;
add(a,b,-c);
}
else
{
int a,b;cin>>a>>b;
add(a,b,0),add(b,a,0);
}
}
for(int i=1;i<=n;i++) add(0,i,0);
if(spfa(1)) cout<<"Yes";
else cout<<"No";
return 0;
}