【LeetCode：3152】特殊数组 II（Java）

题目链接

• 3152. 特殊数组 II

题目描述

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

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

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

示例 1：

输入：nums = [3,4,1,2,6], queries = [[0,4]]

输出：[false]

解释：

子数组是 [3,4,1,2,6]。2 和 6 都是偶数。

示例 2：

输入：nums = [4,3,1,6], queries = [[0,2],[2,3]]

输出：[false,true]

解释：

子数组是 [4,3,1]。3 和 1 都是奇数。因此这个查询的答案是 false。
子数组是 [1,6]。只有一对：(1,6)，且包含了奇偶性不同的数字。因此这个查询的答案是 true。

提示：

1 <= nums.length <= 105
1 <= nums[i] <= 105
1 <= queries.length <= 105
queries[i].length == 2
0 <= queries[i][0] <= queries[i][1] <= nums.length - 1

求解思路

• 动态规划：验证区间 $[i,j]$ 是否为特殊数组等价于判断原数组以位置 $j$ 结尾的最长特殊数组长度是否比 $j-i$ 大。因此我们通过一个dp[]数组来记录以每个位置结尾的最长特殊数组长度，在依次读取queries[]数组进行判断即可。

实现代码

class Solution {
    public boolean[] isArraySpecial(int[] nums, int[][] queries) {
        int n = nums.length;
        int[] dp = new int[n];
        for (int i = 0; i < n; i ++) {
            dp[i] = 1;
        }
        for (int i = 1; i < n; i ++) {
            if (((nums[i] ^ nums[i-1]) & 1) == 1) {
                dp[i] = dp[i - 1] + 1;
            }
        }

        boolean[] res = new boolean[queries.length];
        for (int i = 0; i < queries.length; i ++) {
            int j = queries[i][0], k = queries[i][1];
            res[i] = k - j < dp[k];
        }

        return res;
    }
}