『题解』洛谷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
3 3
3 1 2
1 1 3 1
2 2 3 2
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;
}
