力扣3152.特殊数组Ⅱ随笔

“尽信书，则不如无书。”——《孟子》

题目

如果数组的每一对相邻元素都是两个奇偶性不同的数字，则该数组被认为是一个 特殊数组 。

周洋哥有一个整数数组 nums 和一个二维整数矩阵 queries，对于 queries[i] = [fromi, toi]，请你帮助周洋哥检查子数组 nums[fromi..toi] 是不是一个 特殊数组 。

返回布尔数组 answer，如果 nums[fromi..toi] 是特殊数组，则 answer[i] 为 true ，否则，answer[i] 为 false 。

难度：中等

分析

        笔者自以为找到了很好的方法，但没有想到最显而易见的方法🤦‍，在此记录两种方法。

        笔者的思路受力扣3244. 新增道路查询后的最短距离 II解答的影响，将完整的数组分隔成一个个特殊数组，并且用数组记录每一个连续数组的左端点。对于每一次查询[from,to]，在记录数组中找到小于等于from的左端点，判断下一个值是否大于to。

        此题比较方便的方法是动态规划。用dp[i]表示以i为右端点的连续数组的最大长度，对于每一次查询[from,to]，dp[to]大于等于to-from+1则满足条件。状态转移方程很简单：nums[i]与nums[i-1]奇偶性不同则dp[i]=dp[i-1]+1，否则dp[i]=1。

解答

二分查找

class Solution {
public:
    vector<bool> isArraySpecial(vector<int>& nums, vector<vector<int>>& queries) {
        int n=nums.size(),m=queries.size();
        vector<bool> ans(m);
        vector<int> arr={0};
        for (int i=1;i<n;i++){
            if (nums[i]%2==nums[i-1]%2){
                arr.push_back(i);
            }
        }
        for (int i=0;i<m;i++){
            int from=queries[i][0],to=queries[i][1];
            auto iter=upper_bound(arr.begin(),arr.end(),from);
            int check=iter-arr.begin();
            if (check==arr.size()||arr[check]>to){
                ans[i]=true;
            }else{
                ans[i]=false;
            }
        }
        return ans;
    }
};

动态规划

class Solution {
public:
    vector<bool> isArraySpecial(vector<int>& nums, vector<vector<int>>& queries) {
        int n=nums.size(),m=queries.size();
        vector<bool> ans(m,false);
        vector<int> dp(n,1);
        for (int i=1;i<n;i++){
            if (nums[i]%2!=nums[i-1]%2){
                dp[i]=dp[i-1]+1;
            }
        }
        for (int i=0;i<m;i++){
            int from=queries[i][0];
            int to=queries[i][1];
            if (dp[to]>=to-from+1){
                ans[i]=true;
            }
        }
        return ans;
    }
};