每周题解：Intervals（区间）

题目链接

区间

题目描述

给定 $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

样例输入 #1

5
3 7 3
8 10 3
6 8 1
1 3 1
10 11 1

样例输出 #1

6

提示

【数据范围】

$1\le n\le 50000$,
$0\le a_i,b_i\le 50000$,
$0\le c_i\le b_i-a_i+1$

算法思想

根据题目描述，给定 $n$ 个区间 $[a_i,b_i]$，从每个区间至少 $c_i$ 个数构造一个整数集合 $Z$，求集合 $Z$ 最少包含多少个数？测试样例中有$5$个区间：

• $[3,7]$中至少选$3$个数，可以选：$3,4,5$
• $[8,10]$中至少选$3$个数，可以选：$8,9,10$
• $[6,8]$中至少选$1$个数，可以选：$8$
• $[1,3]$中至少选$1$个数，可以选：$3$
• $[10,11]$中至少选$1$个数，可以选：$10$

集合 $Z$ 最少包含$6$个数：$3,4,5,8,9,10$。

这里可以参考前缀和的思想，用$s[i]$表示区间$[1,i]$中选取数的个数，可以得到下面的一系列关系：

• 从前$i$个数中选取数的个数一定不少于从前$i-1$个数中选，$s[i]\ge s[i-1]$
• 从前$i$个数中取比从前$i-1$个数中取、最多会多$1$个$，s[i]-s[i-1]\le 1\Rightarrow s[i-1]\ge s[i]-1$
• 对于区间$[a,b]$中至少取$c$个数，那么$s[b]-s[a-1]\ge c\Rightarrow s[b]\ge s[a-1]+c$

根据上述一系列不等式关系，可以用差分约束的思想来求集合 $Z$ 中最少包含数的个数。差分约束可以参考博主的另一篇文章：每周算法：差分约束。

具体过程如下：

• 对于$\forall i\in [1,n]$，存在不等式$s[i]\ge s[i-1]$，从节点$i-1$到$i$建一条有向边，边权为$0$
• 对于$\forall i\in [1,n]$，存在不等式$s[i-1]\ge s[i]-1$，从节点$i$到$i-1$建一条有向边，边权为$-1$
• 对于区间$[a,b]$中至少取$c$个数，存在不等式$s[b]\ge s[a-1]+c$，从节点$a-1$到$b$建一条有向边，边权为$c$
• 建立一个虚拟源点$0$，$s[0]$表示在前$0$个数中选取$0$个点。
• 从虚拟源点$0$出发求单源最长路。由于区间中每个点都有边相连，因此从虚拟源点$0$出发可以访问到图中每个点，也就能访问每条边，满足差分约束系统中的不等式组。
• 最终$s[T]$就是集合 $Z$ 中最少包含数的个数，这里$T$指的是所有区间中最后一个数。

注意：

• 由于$0\le c_i\le b_i-a_i+1$，说明最终问题一定有解，所以不用判断是否存在正环，最坏情况把区间中所有数都选上。
• 由于$0\le a_i,b_i\le 50000$，区间开始位置包含$0$点，而做题中需要建立一个$0$作为虚拟源点，因此需要对所有区间右移一个数，避免取到$0$点。

代码实现

#include <bits/stdc++.h>
using namespace std;
const int N = 50005, M = N * 3 + 5;
int h[N], e[M], w[M], ne[M], idx;
int n, T, s[N], q[N];
bool st[N];
void add(int a, int b, int c)  // 添加一条边a->b，边权为c
{
    e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx ++ ;
}

void spfa()
{
    memset(s, -0x3f, sizeof h);
    int hh = 0, tt = 0;
    q[tt ++] = 0, s[0] = 0, st[0] = true; //虚拟源点入队
    while(hh != tt)
    {
        int u = q[hh ++];
        if(hh == N) hh = 0;
        st[u] = false;
        for(int i = h[u]; ~ i; i = ne[i])
        {
            int v = e[i];
            if(s[v] < s[u] + w[i]) //求最长路
            {
                s[v] = s[u] + 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 < N; i ++) //注意，这里要对区间中每个点建边
    {
        add(i - 1, i, 0); //s[i]>=s[i-1]
        add(i, i - 1, -1); //s[i-1]>=s[i]-1
    }
    for(int i = 1; i <= n; i ++)
    {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        a ++, b ++; //区间整体向右移动
        add(a -1, b, c); //s[b]>=s[a-1]+c
        T = max(T, b);
    }
    spfa(); //求差分约束
    printf("%d", s[T]); //输出前T个数中最少包含多少个数
    return 0;
}