【ACWing】4195. 线段覆盖

题目地址：

https://www.acwing.com/problem/content/description/4198/

在一个坐标轴上有$n$条线段。每条线段的每个端点的坐标都为整数。可能存在退化成点的线段。线段之间可以相互交叉、嵌套甚至重合。请你计算，对于每个$k\in 1,2,\dots ,n$，坐标轴中共有多少个整数坐标的点满足恰好被$k$条线段覆盖。注意，左右端点分别为$l_i,r_i$的线段覆盖点$x$当且仅当$l_i\le x\le r_i$。

输入格式：
第一行包含整数$n$。接下来$n$行，每行包含两个整数$l_i,r_i$，表示一条线段的左右端点。

输出格式：
一行$n$个整数，其中第$i$个整数表示坐标轴中满足恰好被$i$条线段覆盖的整数坐标的点的数量。

数据范围：
前三个测试点满足$1\le n\le 3$。
所有测试点满足$1\le n\le 2\times 10^5$，$0\le l_i\le r_i\le 10^{18}$。

可以用前缀和。由于区间范围比较大，所以需要用map（C++里的map是用红黑树实现的，是有序的）。接着遍历map，维护上个pair的端点位置$l$，和上个pair对应的前缀和$s$。在遍历到$(k,v)$的时候，如果这个pair不是第一个的话，那么说明$l,l+1,...,k-1$这$k-l$个数出现的次数是$s$，统计一下即可。接着更新$l$和$s$。代码如下：

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

const int N = 2e5 + 10;
int n;
long cnt[N];
map<long, int> b;

int main() {
  scanf("%d", &n);
  for (int i = 1; i <= n; i++) {
    long l, r;
    scanf("%ld%ld", &l, &r);
    b[l]++, b[r + 1]--;
  }

  long sum = 0, last = -1;
  for (auto& [k, v] : b) {
    if (~last) cnt[sum] += k - last;
    sum += v;
    last = k;
  }

  for (int i = 1; i <= n; i++) printf("%ld ", cnt[i]);
}

时空复杂度$O(n)$。