『题解』洛谷P1993 小K的农场

更好的阅读体验

Portal

Portal1: Luogu

Description

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

农场$a$比农场$b$至少多种植了$c$个单位的作物，
农场$a$比农场$b$至多多种植了$c$个单位的作物，
农场$a$与农场$b$种植的作物数一样多。

但是，由于小$K$的记忆有些偏差，所以他想要知道存不存在一种情况，使得农场的种植作物数量与他记忆中的所有信息吻合。

Input

第一行包括两个整数$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$一样多。

Output

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

Sample Input

Sample Output

Yes

Hint

对于$100\%$的数据保证：$1 \le n, m, a, b, c \le 10000$。

Solution

我们用$\mathrm{s[i]}$表示$i$农场的作物数量，那么题目中的条件我们可以表示为：

$s[a] \ge s[b] + c$
$s[b] \ge s[a] - c$
$s[b] = s[a]$

因为我们如果想让这道题用差分约束做，要把所有的约束条件都改为$\le$或者$\ge$的形式，但是此题的所有农场的做作物数都不能为负数，所以必须要用最长路解决。因此，我们可以改为：

$s[a] \ge s[b] + c$
$s[b] \ge s[a] - c$
$s[a] \ge s[b] + 0$
$s[b] \ge s[a] + 0$
$s[i] \ge 0$

然后我们开始建边：

对于题目给出的$a, b, c$

$b \to a$建一条权值为$c$的边；
$a \to b$建一条权值为$-c$的边；
$a \to b$建一条权值为$0$的边；
$b \to a$键一条权值为$0$的边；

最后在$0 \to i, i \in [1, n]$建权值为$0$的边。

建完之后用$\mathrm{SPFA}$跑一遍最长路，顺便判断环就可以了。

Code

#include<iostream>

#include<algorithm>

#include<cstdio>

#include<cstring>

#include<cmath>

#include<queue>

using namespace std;

const int INF = 0x3f3f3f3f, MAXN = 20005;

struct EDGE {

    int to, nxt, val;

} edge[MAXN];

int n, m, u, v, opt, val, cnt, tot[MAXN], vis[MAXN], dis[MAXN], head[MAXN];

inline void addedge(int u, int v, int val) {//邻接表存图

    edge[++cnt].to = v; edge[cnt].val = val; edge[cnt].nxt = head[u]; head[u] = cnt;

}

inline bool SPFA() {//SPFA最长路

    priority_queue<int> Q;

    memset(dis, -INF, sizeof(dis));

    vis[0] = 1;

    dis[0] = 0;

    tot[0] = 1;

    Q.push(0);

    while (!Q.empty()) {

        int u = Q.top();

        Q.pop();

        vis[u] = 0;

        for (int i = head[u]; ~i; i = edge[i].nxt) {

            int v = edge[i].to;

            if (dis[v] < dis[u] + edge[i].val) {

                dis[v] = dis[u] + edge[i].val;

                if (!vis[v]) {

                    tot[v] = tot[u] + 1;

                    Q.push(v);

                    vis[v] = 1;

                    if (tot[v] > n) return 0;

                }

            }

        }

    }

    return 1;

}

int main() {

    scanf("%d%d", &n, &m);

    memset(head, -1, sizeof(head));

    for (int i = 1; i <= m; i++) {

        scanf("%d", &opt);

        if (opt == 1) {

            scanf("%d%d%d", &u, &v, &val);

            addedge(u, v, val);

        } else

        if (opt == 2) {

            scanf("%d%d%d", &u, &v, &val);

            addedge(v, u, -val);

        } else {

            scanf("%d%d", &u, &v);

            addedge(u, v, 0);

            addedge(v, u, 0);

        }

    }

    for (int i = 1; i <= n; i++)

        addedge(0, i, 0);//按题目的描述建边

    if (SPFA()) printf("Yes\n"); else printf("No\n");

    return 0;

}