特殊数组 II----leetcode

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

你有一个整数数组 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]

解释：

1. 子数组是 [4,3,1]。3 和 1 都是奇数。因此这个查询的答案是 false。
2. 子数组是 [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

思路：

1. 构建前缀和数组：

   • 我们需要快速判断一个子数组是否满足“每对相邻元素的奇偶性不同”。通过构建一个前缀和数组 s，我们可以记录从数组的开始到每个位置为止，有多少对相邻元素的奇偶性相同。
   • 具体来说，如果 nums[i] 和 nums[i+1] 奇偶性相同，那么 s[i+1] = s[i] + 1，否则 s[i+1] = s[i]。

2. 处理查询：

   • 对于每个查询 [fromi, toi]，我们可以通过 s 数组来快速计算在这个区间内是否存在奇偶性相同的相邻元素。如果存在，那么 s[toi] - s[fromi] 将大于0，此时该子数组不是特殊数组。

代码实现：

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
bool* isArraySpecial(int* nums, int numsSize, int** queries, int queriesSize, int* queriesColSize, int* returnSize) {
    // 创建前缀和数组 s，用于记录奇偶性相同的相邻元素对数
    int *s = malloc(sizeof(int) * numsSize);
    memset(s, 0, sizeof(int) * numsSize);
    
    // 构建前缀和数组
    for(int i = 0; i < numsSize - 1; i++) {
        // 如果相邻元素的奇偶性相同，则前缀和数组增加1
        s[i+1] = s[i] + (((nums[i] % 2) == (nums[i+1] % 2)) ? 1 : 0);
    }

    *returnSize = queriesSize;
    bool *ans = malloc(sizeof(bool) * queriesSize);

    // 处理每一个查询
    for(int i = 0; i < queriesSize; i++) {
        // 如果在范围内有奇偶性相同的相邻元素，则返回 false
        ans[i] = (((s[queries[i][1]] - s[queries[i][0]]) > 0) ? false : true);
    }
    
    // 返回结果数组
    return ans;
}

程序执行结果：

代码注释与解释：

1. 前缀和数组 s 的初始化：

int *s = malloc(sizeof(int) * numsSize);
memset(s, 0, sizeof(int) * numsSize);

• 分配内存用于存储前缀和数组 s，并将其初始化为0。

2. 构建前缀和数组：

for(int i = 0; i < numsSize - 1; i++) {
    s[i+1] = s[i] + (((nums[i] % 2) == (nums[i+1] % 2)) ? 1 : 0);
}

• 逐一检查相邻元素 nums[i] 和 nums[i+1] 是否奇偶性相同。如果相同，s[i+1] 的值增加1，否则保持不变。

3. 处理查询：

for(int i = 0; i < queriesSize; i++) {
    ans[i] = (((s[queries[i][1]] - s[queries[i][0]]) > 0) ? false : true);
}

• 对于每个查询 [fromi, toi]，通过检查 s[toi] - s[fromi] 是否大于0，来判断在该子数组内是否存在奇偶性相同的相邻元素。
• 如果存在，则返回 false，否则返回 true。

4. 返回结果数组

• 最后，将布尔数组 ans 返回，供调用者使用。

题目来源：https://leetcode.cn/problems/special-array-ii/

思路来源：b站灵神