【图论】—— 差分约束系统

玄澈_

已于 2022-06-15 17:19:09 修改

阅读量204

点赞数 4

分类专栏： ACM算法笔记 # ACM算法-图论文章标签：图论图搜索算法 c++ 蓝桥杯算法

于 2022-06-15 15:51:06 首次发布

本文链接：https://blog.csdn.net/forever_bryant/article/details/125296144

版权

ACM算法笔记同时被 2 个专栏收录

31 篇文章 8 订阅

订阅专栏

ACM算法-图论

7 篇文章 1 订阅

订阅专栏

差分约束

差分约束系统是一种特殊的N元一次不等式组。它包含N个变量 $\small X_{1} \sim X_{N}$ ，以及 N 个约束条件，

每个约束条件都是由两个变量作差构成的，形如 $\large X_{i}-X_{j}\leq c_{k}$ ，其中 $c_{k}$ 可以是常数（可以是非负数，也可以是负数）， $\small 1\leq i,j\leq N, 1\leq k\leq M$ 。

我们要解决的问题就是：求一组解 $\small X_{1} = a_{1},X_{2} = a_{2},\cdots X_{N} = a_{N}$ ，使得所有的约束条件都得到满足。

差分约束系统中的每个约束条件 $X_{i}-X_{j}\leq c_{k}$ 可以变形为 $\large X_{i}\leqslant X_{j}+c_{k}$ 。这和单源最短路问题中的三角不等式 $\large dist[y]<dist[x]+z$ 十分相似。因此，可以把每个变量 $X_{i}$ 看做是有向图中的一个结点 i ，对于每个约束条件 $X_{i}-X_{j}\leq c_{k}$ ，从结点 j 向结点 i 连一条长度为 $c_{k}$ 的有向边。

算法思路

“超级源点”

注意到如果 $\left \{ \left. a_{1},a_{2},\cdots a_{n} \right \} \right.$ 是一组解，那么对于任意的常数 $\Delta$ ， $\left \{ \left. a_{1} + \Delta ,a_{2}+ \Delta ,\cdots a_{n}+ \Delta \right \} \right.$ 显然也是一组解（作差后 $\Delta$ 恰好被消掉）。所以不妨求解一组负数解，即假设 $\forall i,X_i \leqslant 0$ ，然后再增加一个 0 号结点。令 $X_0 = 0$ 。

这样一来，就多了 N 个形如 $X_i - X_0 \leqslant 0$ 的约束条件，应该从节点 0 向每个节点 i 连一条长度为0的有向边。

设 $dist[0]$ 等于0，设 0 为起点求单源最短路。若图中存在负环，则给定的差分约束系统无解。否则， $X_i=dist[i]$ 就是差分约束系统的一组解。

例题讲解：AcWing 1169. 糖果

幼儿园里有 NN 个小朋友，老师现在想要给这些小朋友们分配糖果，要求每个小朋友都要分到糖果。

但是小朋友们也有嫉妒心，总是会提出一些要求，比如小明不希望小红分到的糖果比他的多，于是在分配糖果的时候，老师需要满足小朋友们的 KK 个要求。

幼儿园的糖果总是有限的，老师想知道他至少需要准备多少个糖果，才能使得每个小朋友都能够分到糖果，并且满足小朋友们所有的要求。

输入格式

输入的第一行是两个整数 N,K

接下来 KK 行，表示分配糖果时需要满足的关系，每行 3 个数字 X,A,B

如果 X=1X=1．表示第 A 个小朋友分到的糖果必须和第 B 个小朋友分到的糖果一样多。
如果 X=2X=2，表示第 A 个小朋友分到的糖果必须少于第 B 个小朋友分到的糖果。
如果 X=3X=3，表示第 A 个小朋友分到的糖果必须不少于第 B 个小朋友分到的糖果。
如果 X=4X=4，表示第 A 个小朋友分到的糖果必须多于第 B 个小朋友分到的糖果。
如果 X=5X=5，表示第 A 个小朋友分到的糖果必须不多于第 B 个小朋友分到的糖果。

小朋友编号从 11 到 NN。

输出格式

输出一行，表示老师至少需要准备的糖果数，如果不能满足小朋友们的所有要求，就输出 −1。

数据范围

1≤N<1051≤N<105,
1≤K≤1051≤K≤105,
1≤X≤51≤X≤5,
1≤A,B≤N1≤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

解题思路

AC代码

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

typedef long long LL;

const int N = 100010, M = 300010;

int n, m;
int h[N], e[M], w[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[ -- tt];
        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;
                    st[j] = true;
                }
            }
        }
    }

    return 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(b, a, 0), add(a, b, 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
    {
        LL res = 0;
        for (int i = 1; i <= n; i ++ ) res += dist[i];
        printf("%lld\n", res);
    }

    return 0;
}

例题讲解：AcWing 362. 区间

解题思路

AC代码

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 50010, M = 150010;

int n;
int h[N], e[M], w[M], ne[M], idx;
int dist[N];
int q[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 ++ ;
}

void spfa()
{
    memset(dist, -0x3f, sizeof dist);
    dist[0] = 0;
    st[0] = true;
    int hh = 0, tt = 1;
    q[0] = 0;

    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];
                if (!st[j])
                {
                    q[tt ++ ] = j;
                    if (tt == N) tt = 0;
                    st[j] = true;
                }
            }
        }
    }
}

int main()
{
    scanf("%d", &n);

    memset(h, -1, sizeof h);
    for (int i = 1; i < N; i ++ )
    {
        add(i - 1, i, 0);
        add(i, i - 1, -1);
    }

    for (int i = 0; i < n; i ++ )
    {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        a ++, b ++ ;
        add(a - 1, b, c);
    }

    spfa();

    printf("%d\n", dist[50001]);

    return 0;
}