Acwing_1169糖果【差分约束】

题目描述：

幼儿园里有 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;
}