AcWing 101. 最高的牛 (差分+前缀和)

AcWing 101. 最高的牛 (差分+前缀和)

有 N 头牛站成一行，被编队为 1、2、3…N，每头牛的身高都为整数。

当且仅当两头牛中间的牛身高都比它们矮时，两头牛方可看到对方。

现在，我们只知道其中最高的牛是第 P 头，它的身高是 H ，剩余牛的身高未知。

但是，我们还知道这群牛之中存在着 M 对关系，每对关系都指明了某两头牛 A 和 B 可以相互看见。

求每头牛的身高的最大可能值是多少。

输入格式

第一行输入整数 N,P,H,M，数据用空格隔开。

接下来 M 行，每行输出两个整数 A 和 B ，代表牛 A 和牛 B 可以相互看见，数据用空格隔开。

输出格式

一共输出 N 行数据，每行输出一个整数。

第 i 行输出的整数代表第 i 头牛可能的最大身高。

数据范围

1 ≤ N ≤ 10000,
1 ≤ H ≤ 1000000,
1 ≤ A, B ≤ 10000,
0 ≤ M ≤ 10000

输入样例：

9 3 5 5
1 3
5 3
4 3
3 7
9 8

输出样例：

5
4
5
3
4
4
5
5
5

题解：

可以假设一开始所有的牛高度都是0，如果我们直到A牛和B牛可以相互看见，那么就说明这两头牛之间的牛都比它们矮，我们可以看作将A + 1到B - 1的牛高度都 - 1，这样处理完所有的AB牛关系后，在将所有的牛高度加上已知最高的牛高度h，这样就可以得到符合题意的牛的身高了。

我们需要将区间内的数统减一，如果采用暴力的方法一个个遍历减一，这样时间复杂度是O(n)，会超时。
我们可以利用差分的知识点，在区间[l, r]内每个数-1，就是将差分数组b[l + 1] - 1，再将b[r + 1] + 1。这样可以在O(n)复杂度下修改区间值。
处理完差分数组后我们再将差分数组求前缀和就可以恢复原数组，再将原数组每个数+h即可得到答案。

注意题目给的AB关系可能会有重复的，用set去重即可，不能重复处理。

AC代码:

#include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 5; 
int d[N]; // 差分数组 
int main() {
    ios::sync_with_stdio(false);
    int n, p, h, m;
    cin >> n >> p >> h >> m;
    set<pair<int, int> > st; // 去除重复的AB关系 
    while (m --) {
        int a, b;
        cin >> a >> b;
        if (a > b) swap(a, b); // 保证A小B大 
        if (st.count({a, b}) > 0) continue;
        else st.insert({a, b});
        d[a + 1] --; // 利用差分将A + 1和B - 1区间都减一 
        d[b] ++;
    }
    d[1] += h; // 求前缀和后所有牛高度加h 
    for (int i = 1; i <= n; i ++) { // 差分数组求前缀和恢复原数组 
        d[i] += d[i - 1];
        cout << d[i] << '\n'; // 输出答案 
    }
    return 0;
}