洛谷P1993 小K的农场

时间:2023-03-08 16:50:33
洛谷P1993 小K的农场

思路是差分约束+dfs版SPFA。

首先来思考差分约束的过程,将题目给出的式子进行转化:

  • 农场a比农场b至少多种植了c个单位的作物,

SPFA我们考虑跑最短路,那么要让SPFA中满足的式子就是if(d[b]>d[a]-c)d[b]=d[a]-c,即让b<=a-c。

所以建一条a->b权值为-c的边,使d[b]始终满足<=d[a]-c。

  • 农场a比农场b至多多种植了c个单位的作物,

同理,建一条b->a,权值为c的边。

  • 农场a与农场b种植的作物数一样多。

建一条a,b间的双向边,权值为0。即建两条单向权值为0的边。

最后处理完所有条件,考虑存在整张图不连通的情况——这时候我们需要一个点能把整张图串起来,又不会影响到结果,即“超级源点”0点。

由0点向每个点建一条权值为0的边。

然后就是SPFA松弛,判负环的过程。这里用dfs版的SPFA不然会T。

代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
int n,m,d[],vis[],cnt[],flag;
int ver[],Next[],edge[],head[],tot;
queue<int>q;
void add(int a,int b,int c){
ver[++tot]=b;
Next[tot]=head[a];
edge[tot]=c;
head[a]=tot;
}
int spfa(int x){
vis[x]=;
for(int i=head[x];i;i=Next[i]){
int v=ver[i],z=edge[i];
if(d[v]>d[x]+z){
d[v]=d[x]+z;
if(vis[v])return ;
if(!spfa(v))return ;
}
}
vis[x]=;
return ;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=,k,a,b,c;i<=m;i++){
scanf("%d",&k);
if(k==){
scanf("%d%d%d",&a,&b,&c);
add(a,b,-c);
}
if(k==){
scanf("%d%d%d",&a,&b,&c);
add(b,a,c);
}
if(k==){
scanf("%d%d",&a,&b);
add(a,b,);
add(b,a,);
}
}
for(int i=;i<=n;i++)add(,i,);
memset(d,0x3f,sizeof(d));
d[]=;
if(!spfa())printf("No");
else printf("Yes");
return ;
}