AcWing 1169 糖果

    科技2025-12-27  10

    题目描述:

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

    分析:

    本题考查差分约束。如果一个系统由n个变量和m个约束条件组成,其中每个约束条件形如xj-xi<=bk(i,j∈[1,n],k∈[1,m]),则称其为差分约束系统(system of difference constraints)。亦即,差分约束系统是求解关于一组变量的特殊不等式组的方法。简而言之,差分约束就是用来求解一种特殊的不等式组,这种不等式组里面不等式的格式都是类似于Xa <= Xb + c这种形式,差分约束问题可以用求单源最短路的算法来求解,一般使用spfa算法求解。

    1.差分约束问题与最短路问题的联系

    在求解最短路的算法中,核心的语句就是松弛操作,即d[v] > d[u] + w,就可以更新d[v]为d[u] + w了。在松弛某点v的距离的时候,会考察所有可以到达v的相邻点,每次从v的相邻点尝试去松弛v之后,都会有d[v] <= d[u] + w,所以最终最短路径上d[v] = min(d[i] + w[i])。如果将d[v]看作Xa,所有与之相邻的点看作Xb,则求完最短路径后的{Xa,Xb}的值一定满足形如Xa <= Xb + c的不等式组的要求,也就是说,可以用spfa求最短路的过程来求解差分约束问题。

    2.不等式组解的最小值与最大值

    在求最短路的过程中,如果起点s的距离设为0,设从起点到某点t的路径上依次经过x1,x2,x3三个点,对应的边权依次是c1,c2,c3。则有x1 <= s + c1,x2 <= x1 + c2,x3 <= x2 + c3,可以推出x3 <= x2 + c3 <= x1 + c2 + c3 <= s + c1 + c2 + c3,s在这里是定值,可以看出x3的值不会超过s + c1 + c2 + c3,也就是可以找到x3的最大值,但是不一定能够知道x3的最小值,其他变量x1,x2也均有个上限,所以求最短路得到的解实际上是不等式组的最大解。当然,在松弛过程中d[v] = min(d[i] + w[i]),也就是为了满足所有的约束条件,d[v]实际上是取所有松弛结果的最小值的。

    再来考察求最长路的过程,将求最短路的松弛条件倒过来就可以求最长路了,即d[v] < d[u] + w时,就更新d[v]为d[u] + w,最后d[v] = max(d[i] + w[i]),也就是求完最长路后d[v] >= d[u] + w。同样的如果起点s的距离设为0,设从起点到某点t的路径上依次经过x1,x2,x3三个点,对应的边权依次是c1,c2,c3。则有x1 >= s + c1,x2 >= x1 + c2,x3 >= x2 + c3,可以推出x3 >= x2 + c3 >= x1 + c2 + c3 >= s + c1 + c2 + c3,s在这里是定值,可以看出x3的值不会小于s + c1 + c2 + c3,这样就求出了x3的最小值了,也就是说最长路可以求出不等式组的最小解。注意,最长路算法求解的是形如xa >= xb + c这种形式的不等式。

    3.无约束的变量与无解情况

    假设图中有一点是孤立的,与其他点没有关联边,则对应差分约束问题中的变量就是不受约束的,可以取任意值。再来考察无解的情况,如果求最短路时存在负环,假设负环上的点有x1,x2,x3,则有x2 <= x1 + c1,x3 <= x2 + c2,x1 <= x3 + c3,可以推出x1 <= x3 + c3 <= x2 + c2 + c3 <= x1 + c1 + c2 + c3,又c1 + c2 + c3 < 0(存在负环),所以x1 <= x1 + c1 + c2 + c3 < x1得出了x1 < x1的矛盾结论了,所以spfa算法如果求出了负环则说明差分约束问题无解。

    求最长路时如果存在正环,设正环上的点有x1,x2,x3,则有x2 >= x1 + c1,x3 >= x2 + c2,x1 >= x3 + c3,可以推出x1 >= x3 + c3 >= x2 + c2 + c3 >= x1 + c1 + c2 + c3,又c1 + c2 + c3 > 0(存在正环),所以x1 >= x1 + c1 + c2 + c3 > x1得出了x1 > x1的矛盾结论了。

    4.差分约束问题的建图

    找到了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使用队列会超时,所以要用栈替换队列。

    #include <iostream> #include <cstdio> #include <cstring> #include <algorithm> using namespace std; const int N = 100005,M = 300005; int idx,h[N],e[M],w[M],ne[M]; bool st[N]; int n,k,d[N],q[N],cnt[N]; void add(int a,int b,int c){ e[idx] = b,w[idx] = c,ne[idx] = h[a],h[a] = idx++; } bool spfa(){ memset(d,-0x3f,sizeof d); memset(st,false,sizeof st); memset(cnt,0,sizeof cnt); int hh = 0; q[hh++] = 0; d[0] = 0; st[0] = true; while(hh){ int u = q[--hh]; st[u] = false; for(int i = h[u];~i;i = ne[i]){ int j = e[i]; if(d[j] < d[u] + w[i]){ d[j] = d[u] + w[i]; cnt[j] = cnt[u] + 1; if(cnt[j] >= n + 1) return false; if(!st[j]){ q[hh++] = j; st[j] = true; } } } } return true; } int main(){ scanf("%d%d",&n,&k); int a,b,x; memset(h,-1,sizeof h); for(int i = 0;i < k;i++){ 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); if(!spfa()) puts("-1"); else{ long long res = 0; for(int i = 1;i <= n;i++){ res += d[i]; } printf("%lld\n",res); } return 0; }

     

    Processed: 0.052, SQL: 9