[M前缀和] lc3152. 特殊数组 II(前缀和变种+二分+lower_bound操作+好题)

1. 题目来源

链接：3152. 特殊数组 II

关于一些常用的 STL 函数：

• [杂谈] 3.#include algorithm 用法用途

2. 题目解析

本题思路比较明确哈。首先自己想的是二分的思路，但碍于一些细节，代码没有写出来，但思路没啥问题。看题解用的是前缀和的思路，是正解，更加简单直接。官解中用的 dp，没啥必要在这里。

---

二分思路：

• 预先统计不满足要求的数对的左下标位置。
• 对于每一个查询 q，记 l=q[0]，此时可以二分查找第一个大于等于 l 的不满足下标的位置，记为 t。
• 当查询的右边界 r=q[0]
  • 有 r<=t 时，说明右边界并没有延伸过这个不满足的数对，说明结果有效，添加 true 即可。
  • 否则，说明右边界延伸过这个不满足的数对了，结果无效，添加 false。
• 需要注意的是，可以向右边界添加一个哨兵极大值，这样就能一直找到大于等于 l 的位置了。
• 关于整数二分，还能取到边界的情况下，可以直接使用 lower_bound 操作简化代码。

---

前缀和思路：

• 对于这种区间询问的，可以使用前缀和优化的思路。
• 对于原数组中，每一个不满足的数对，都认为他是一个端点，将断点记为 1。
• 计算前缀和。
• 对于查询，可以获取 l,r 之前的前缀和是否为 0，判断是否存在断点，存在则不满足。记录答案即可。

但对于前缀和这里有个点需要注意下：

• 累计前缀和的时候，数对下标 [0, 1] 如果不满足的话，这个将累计到 f[1] 位置的前缀和数组中。[1,2] 不满足的话，将累计到 f[2] 的数组中。
• 如果查询来的是 [1,2] 区间的话，那么我们直接使用 f[2]-f[1] 即可，而不是 f[2]-f[1-1] 这种前缀和的处理操作。
• 其实在我理解来看，这里应该是做的 [l, r] 区间，应该是 f[r] - f[l+1-1] 这种操作。因为 f[l+1] 才是 [l+1, l] 的数值是否满足条件的贡献值。-1 是为了能够取到 f[l+1] 这个贡献。所以导致了 f[l] 这种纯下标的写法。

---

明显，前缀和思路考虑问题更加少，代码也十分好写。

---

• 时间复杂度：$O(n)$
• 空间复杂度：$O(1)$

---

普通整数二分

class Solution {
public:
    vector<bool> isArraySpecial(vector<int>& nums, vector<vector<int>>& queries) {
        int n = nums.size();

        vector<int> sp; 
        // sp.push_back(-1);
        for (int i = 1; i < n; i ++ ) {
            int a = nums[i - 1], b = nums[i];
            if (a % 2 == b % 2) 
                sp.push_back(i - 1);
        }
        sp.push_back(1e9);

        for (int i : sp) cout << i << endl;
        cout << endl;

        int lsp = sp.size();
        vector<bool> res;
        for (auto& q : queries) {
            int a = q[0], b = q[1];
            int l = 0, r = lsp - 1;
            while (l < r) {
                int mid = l + r >> 1;
                if (sp[mid] >= a) r = mid;
                else l = mid + 1;
            }
            cout << l << endl;
            
            if (b <= sp[l]) res.push_back(true);
            else res.push_back(false);
        }

        return res;
    }
};

lower_bound 操作：

class Solution {
public:
    vector<bool> isArraySpecial(vector<int>& nums, vector<vector<int>>& queries) {
        int n = nums.size();

        vector<int> sp; 
        for (int i = 1; i < n; i ++ ) {
            int a = nums[i - 1], b = nums[i];
            if (a % 2 == b % 2) 
                sp.push_back(i - 1);
        }
        sp.push_back(1e9);


        int lsp = sp.size();
        vector<bool> res;
        for (auto& q : queries) {
            int a = q[0], b = q[1];
            // int l = 0, r = lsp - 1;
            // while (l < r) {
            //     int mid = l + r >> 1;
            //     if (sp[mid] >= a) r = mid;
            //     else l = mid + 1;
            // }

            auto p = lower_bound(sp.begin(), sp.end(), a);
            
            if (b <= *p) res.push_back(true);
            else res.push_back(false);
        }

        return res;
    }
};

前缀和操作：

class Solution {
public:
    vector<bool> isArraySpecial(vector<int>& nums, vector<vector<int>>& queries) {
        int n = nums.size();
        vector<int> f(n);

        for (int i = 1; i < n; i ++ ) f[i] = f[i - 1] + (nums[i] % 2 == nums[i - 1] % 2);

        vector<bool> res;
        for (auto& q : queries) res.push_back(f[q[1]] - f[q[0]] == 0);
        
        return res;
    }
};