题目描述:
幼儿园里有 N 个小朋友,老师现在想要给这些小朋友们分配糖果,要求每个小朋友都要分到糖果。
但是小朋友们也有嫉妒心,总是会提出一些要求,比如小明不希望小红分到的糖果比他的多,于是在分配糖果的时候, 老师需要满足小朋友们的 K 个要求。
幼儿园的糖果总是有限的,老师想知道他至少需要准备多少个糖果,才能使得每个小朋友都能够分到糖果,并且满足小朋友们所有的要求。
输入格式:
输入的第一行是两个整数 N,K。
接下来 K 行,表示分配糖果时需要满足的关系,每行 3 个数字 X,A,B。
- 如果 X=1.表示第 A 个小朋友分到的糖果必须和第 B 个小朋友分到的糖果一样多。
- 如果 X=2,表示第 A 个小朋友分到的糖果必须少于第 B 个小朋友分到的糖果。
- 如果 X=3,表示第 A 个小朋友分到的糖果必须不少于第 B 个小朋友分到的糖果。
- 如果 X=4,表示第 A 个小朋友分到的糖果必须多于第 B 个小朋友分到的糖果。
- 如果 X=5,表示第 A 个小朋友分到的糖果必须不多于第 B 个小朋友分到的糖果。
小朋友编号从 1 到 N。
输出格式:
输出一行,表示老师至少需要准备的糖果数,如果不能满足小朋友们的所有要求,就输出 −1。
数据范围:
1≤N<10^5,
1≤K≤10^5,
1≤X≤5,
1≤A,B≤N
输入样例:
5 7
1 1 2
2 3 2
4 4 1
3 4 5
5 4 5
2 3 5
4 5 1
输出样例:
11
思路分析:
差分约束一些用法(具体讲解可参考 《算法竞赛进阶指南》 这本书):
差分约束问题的建图
-
找到了spfa算法可以求解差分约束问题,下面需要做的就是将不等式组转化为图。建图的过程深刻的反映了求最短路最长路与差分约束问题的关联。比如x1 <= x2 + 1,是建一条x2到x1长度为1的边,还是建一条x1到x2长度为-1的边呢?在最短路问题中,我们需要x1 <= x2 + c这种形式的不等式,遇见x1 >= x2 + 1形式的不等式就 转化为了x2 <= x1 - 1,从而建立了x1到x2长度为-1的边。而在最长路问题中,遇见x1 >= x2 + 1可以建一条x2到x1长度为1的边,遇见x1 <= x2 + 1这种形式的不等式可以转化为x2 >= x1 - 1,建立起了x1到x2长度为-1的边。从而得出了一个重要结论:同一个不等式在最长路和最短路问题中建图的方向是相反的,建立的边权互为相反数,比如x1 <= x2 + 1,最短路问题是建一条x2到x1长度为1的边,而在最长路问题中则是建一条x1到x2长度为-1的边(x2 >= x1 - 1)。
-
其他类型的不等式:对于x1 < x2 + c形式的不等式,可以转化为x1 <= x2 + c - 1形式;对于x1 = x2形式的不等式,可以转化为x1 <= x2和x2 <= x1两个不等式。
-
由于建的图不一定连通,所以为了保证从起点出发一定能到达所有点,一般会建一个超级源点,从这个超级源点向各个点引一条长度为0的边,即在不等式组中加上了x0 <= x1,x0 <= x2,…,x0 <=xn这么多不等式。
-
下面回归本题,由于每个小朋友都需要分到糖,所以某个变量都不能小于1,所以建图时可以由x0向各点引一条长度为1的边,因为要求x1 >= 1等价于x1 >= x0 + 1,x0 = 0。
本题是求差分约束的最小解,所以需要求最长路
。spfa算法求最长路时需要将距离数组初始化为负无穷,当存在正环时存在某个点会一直被更新,所以更新超过一定次数后就说明无解。本题的spfa使用队列会超时,所以要用栈替换队列。
代码中我用的还是队列,没有改用栈
AC代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long LL;
const int N = 100010, M = 300010;
int n, m;
int h[N], w[M], e[M], ne[M], idx;
LL dist[N];
int q[N], cnt[N];
bool st[N];
void add(int a, int b, int c) {
e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx++;
}
bool spfa() {
int hh = 0, tt = 1;
memset(dist, -0x3f, sizeof dist);
dist[0] = 0;
q[0] = 0;//超级源点入队
st[0] = true;
while (hh != tt) {//循环队列
int t = q[hh++];
if (hh == N) hh = 0;
st[t] = false;
for (int i = h[t]; ~i; i = ne[i]) {
int j = e[i];
if (dist[j] < dist[t] + w[i]) {//求的是最长路
dist[j] = dist[t] + w[i];
cnt[j] = cnt[t] + 1;
if (cnt[j] >= n + 1) return false;//存在负环
if (!st[j]) {
q[tt++] = j;
if (tt == N) tt = 0;
st[j] = true;
}
}
}
}
}
int main()
{
scanf("%d%d", &n, &m);
memset(h, -1, sizeof h);
while (m--) {
int x, a, b;
scanf("%d%d%d", &x, &a, &b);
if (x == 1) add(a, b, 0), add(b, a, 0);
else if (x == 2) add(a, b, 1);
else if (x == 3) add(b, a, 0);
else if (x == 4) add(b, a, 1);
else add(a, b, 0);
}
for (int i = 1; i <= n; i++) add(0, i, 1);
LL res = 0;
if (!spfa()) puts("-1");
else {
for (int i = 1; i <= n; i++) res += dist[i];
printf("%lld\n", res);
}
system("pause");
return 0;
}