【ACWing】101. 最高的牛

题目地址：

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

有$N$头牛站成一行，被编队为$1$、$2$、$3$…$N$，每头牛的身高都为整数。当且仅当两头牛中间的牛身高都比它们矮时，两头牛方可看到对方。现在，我们只知道其中最高的牛是第$P$头，它的身高是$H$，剩余牛的身高未知。但是，我们还知道这群牛之中存在着$M$对关系，每对关系都指明了某两头牛$A$和$B$可以相互看见。求每头牛的身高的最大可能值是多少。

输入格式：
第一行输入整数$N,P,H,M$，数据用空格隔开。
接下来$M$行，每行输出两个整数$A$和$B$，代表牛$A$和牛$B$可以相互看见，数据用空格隔开。

输出格式：
一共输出$N$行数据，每行输出一个整数。
第$i$行输出的整数代表第$i$头牛可能的最大身高。

数据范围：
$1\le N\le 5000$
$1\le H\le 1000000$
$1\le A,B\le 10000$
$0\le M\le 10000$

首先这$M$对关系所表示的区间是不能有长度大于$0$的交叉的，例如，如果有关系$(a,b),(c,d),a<b,c<d,d>b>c$，那么就会导致$b<c,b>c$，矛盾。我们先假设所有的牛高度都是$H$，如果有关系$(a,b)$，不妨设$a<b$，那么$[a+1,b-1]$这一段里的牛的最高高度就只能是$H-1$了。受此启发，如果某个位置被$k$个关系覆盖，那么这个位置的牛的最高高度就是$H-k$。那么问题就变为，一开始有个数字全为$H$的数组，每个关系就相当于要将$[a+1,b-1]$这一段减$1$，最后求数组变成了什么，可以用差分数组来做。代码如下：

#include <iostream>
#include <unordered_set>
using namespace std;
using PII = pair<int, int>;

const int N = 10010;
int h[N];
auto cmp = [](const PII& p) {
  return hash<int>()(p.first) ^ hash<int>()(p.second);
};
unordered_set<PII, decltype(cmp)> st(0, cmp);

int main() {
  int n, p, H, m;
  scanf("%d%d%d%d", &n, &p, &H, &m);
  h[1] = H;
  for (int i = 0, a, b; i < m; i++) {
    scanf("%d%d", &a, &b);
    if (a > b) swap(a, b);
    if (a + 1 == b) continue;
    if (!st.count({a, b})) {
      st.insert({a, b});
      h[a + 1]--, h[b]++;
    }
  }

  for (int i = 1; i <= n; i++) {
    h[i] += h[i - 1];
    printf("%d\n", h[i]);
  }
}

时间复杂度$O(N+M)$，空间$O(N)$。