【ACWing】362. 区间

题目地址：

https://www.acwing.com/problem/content/364/

给定$n$个区间$[a_i,b_i]$和$n$个整数$c_i$。你需要构造一个整数集合$Z$，使得$\forall i\in [1,n]$，$Z$中满足$a_i\le x\le b_i$的整数$x$不少于$c_i$个。求这样的整数集合$Z$最少包含多少个数。

输入格式：
第一行包含整数$n$。接下来$n$行，每行包含三个整数$a_i,b_i,c_i$。

输出格式：
输出一个整数表示结果。

数据范围：
$1\le n\le 50000$
$0\le a_i,b_i\le 50000$
$1\le c_i\le b_i-a_i+1$

设$s[i]$是在$1\sim 50001$中，前$i$个数中选了多少个数。为了方便我们将所有区间向右平移$1$个单位，使得所有数都位于$[1,50001]$里。那么$s$需要满足以下条件：
1、$s[0]=0$；
2、$s[i]\ge s[i-1],s[i]-s[i-1]\le 1$，意思是$i$这个数最多选$1$次（因为题目里的$Z$是个整数集合，集合是不能有重复元素的）；
3、$s[b_i]-s[a_i-1]\ge c_i$，表示$[a_i,b_i]$这个区间里取的数至少有$c_i$个。
这样就将原问题转化为了一个差分约束问题，要求的就是$s[50001]$的最小值，所以需要将每个不等式变为$s[x]\ge s[y]+z$的形式，然后建图求最长路（从$y$到$x$连一条长$z$的边即可），可以用SPFA来做。代码如下：

#include <iostream>
#include <cstring>
using namespace std;

const int N = 50010, M = 3 * N;
int n;
int h[N], e[M], ne[M], w[M], idx;
int dist[N];
int q[N];
bool st[N];

void add(int a, int b, int c) {
    e[idx] = b, ne[idx] = h[a], w[idx] = c, h[a] = idx++;
}

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

	// 循环队列里要写 while (hh != tt)
    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 v = e[i];
            if (dist[v] < dist[t] + w[i]) {
                dist[v] = dist[t] + w[i];
                if (!st[v]) {
                    q[tt++] = v;
                    if (tt == N) tt = 0;
                    st[v] = true;
                }
            }
        }
    }
    
}

int main() {
    scanf("%d", &n);
    memset(h, -1, sizeof h);
    for (int i = 1; i <= 50001; i++) {
        add(i - 1, i, 0);
        add(i, i - 1, -1);
    }

    while (n--) {
        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;
}

时间复杂度$O(nm)$，空间$O(n)$，$n$和$m$分别是图的点数和边数。