【差分约束】洛谷P1993 小K的农场 题解

【差分约束】洛谷P1993 小K的农场 题解

题目传送门

---

分析

一看题目就知道可以把条件转换成不等式，然后使用差分约束求解。差分约束可以自行 百度OR谷歌。

可以把建边为w(a,b) = c， 转换成 ，建边为w(b,a) = -c。然后使用dfs优化的spfa跑一个最短路判断负环就可以了。

代码

#include <bits/stdc++.h>
#define MAX 100005
using namespace std;

struct Edge {
    int vet, w;
} edge[MAX];

int n, m, cnt;
int head[MAX], Next[MAX];

void add(int x, int y, int z) {
    cnt++;
    Next[cnt] = head[x];
    head[x] = cnt;
    edge[cnt].vet = y;
    edge[cnt].w = z;
}

bool flag, vis[MAX];
int dis[MAX];
void dfs_spfa(int x){
    vis[x] = true;
    for (int i = head[x]; i != 0; i = Next[i]) {
        int v = edge[i].vet, w = edge[i].w;
        if(dis[v] > dis[x]+w){
            if(vis[v]){
                flag = true;
                return;
            }
            dis[v] = dis[x]+w;
            dfs_spfa(v);
        }
    }
    vis[x] = false;
}

int main() {
    int a, b, c, op;
    cin >> n >> m;
    for (int i = 0; i < m; ++i) {
        scanf("%d%d%d", &op, &a, &b);
        if (op == 1) {              //a-b >= c    -->   b-a <= -c
            scanf("%d", &c);
            add(a, b, -c);
        }
        else if (op == 2) {         //a-b <= c
            scanf("%d", &c);
            add(b, a, c);
        }
        else {              //a==b      -->     a-b<=0 && b-a<=0
            add(a,b,0);
            add(b,a,0);
        }
    }
    for (int i = 1; i <= n; ++i) {
        add(0, i, 0);               //以0作为源点
    }
    memset(dis, 0x3f, sizeof(dis));
    dis[0] = 0;
    dfs_spfa(0);
    if(flag){
        cout << "No" << endl;
    }
    else{
        cout << "Yes" << endl;
    }

    return 0;
}