洛谷 P1993 小K的农场 差分约束+dfs_spfa判环

https://www.luogu.org/problem/P1993
题目描述
小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”。

思路：差分约束，如果$a-b>=c$，连一条从$b$到$a$的边权为$c$的有向边，如果$a<=b+c$，做变换后可得$b-a>=-c$，连一条从$a$到$b$的边权为$-c$的有向边，如果$a=b$，则给$a、b$连一条边权为$0$的无向边，那么如果有解，这些约束一定有最小值，即当图中没有正环时，一定有解。用$dfs\_spfa$判断一下就行了，因为图不一定是连通的，所以每个连通块都要跑一次。

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
#define eps 1e-8
#define pr pair<int,int>
using namespace std;
typedef long long ll;

const int maxn=1e4+5;

struct Edge
{
    int to,nxt,dis;
}edge[maxn<<1];

bool vis[maxn],tag[maxn];
int head[maxn],dis[maxn];
int n,m,tot;

bool dfs_spfa(int u)
{
    vis[u]=tag[u]=1;
    for(int i=head[u];i;i=edge[i].nxt)
    {
        if(dis[edge[i].to]<dis[u]+edge[i].dis)
        {
            dis[edge[i].to]=dis[u]+edge[i].dis;
            if(vis[edge[i].to]||dfs_spfa(edge[i].to))
                return 1;
        }
    }
    vis[u]=0;
    return 0;
}

inline void addedge(int u,int v,int dis)
{
    edge[++tot].to=v,edge[tot].nxt=head[u],edge[tot].dis=dis,head[u]=tot;
}

int main()
{
    scanf("%d %d",&n,&m);
    int op,u,v,c;
    for(int i=0;i<m;i++)
    {
        scanf("%d %d %d",&op,&u,&v);
        if(op==1)//u>=v+c -> u-v>=c
        {
            scanf("%d",&c);
            addedge(v,u,c);
        }
        else if(op==2)//u<=v+c -> v-u>=-c
        {
            scanf("%d",&c);
            addedge(u,v,-c);
        }
        else
            addedge(u,v,0),addedge(v,u,0);
    }
    for(int i=1;i<=n;i++)
        dis[i]=-INF;//正环
    bool flag=0;
    for(int i=1;i<=n&&!flag;i++)
        if(!tag[i])//跑完所有的连通块
            dis[i]=0,flag=dfs_spfa(i);
    if(flag)
        printf("No\n");
    else
        printf("Yes\n");
    return 0;
}