【数据结构-前缀和】力扣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。

超时代码

cpp 复制代码
class Solution {
public:
    vector<bool> isArraySpecial(vector<int>& nums, vector<vector<int>>& queries) {
        vector<int> group(nums.size());
        for(int i = 0;i<nums.size();i++){
            group[i] = (nums[i] % 2 == 0) ? 0 : 1;
        }

        return CheckIt(group, queries);
    }

private:
    vector<bool> CheckIt(vector<int>& group, vector<vector<int>>& queries){
        vector<bool> answer;
        for(const auto query : queries){
            bool flag = true;
            for(int i = query[0];i < query[1];i++){
                if(group[i] == group[i+1]){
                    flag = false;
                    break;
                }        
            }
            answer.push_back(flag);
        }
        return answer;
    }
};

group储存偶数为0,奇数为1;

第一次写的代码超时了,问题出在了以下循环

cpp 复制代码
	  	  for(int i = query[0];i < query[1];i++){
              if(group[i] == group[i+1]){
                  flag = false;
                  break;
              }        
          }

我们在处理每次query的时候,根据query给的上限和下限又重新遍历了数组,当query较大的时候,就会消耗很多内存。所以可以考虑是不是可以避免每次计算answer是true还是false的时候,都要把group里面对应的元素循环一遍

优化后的代码

cpp 复制代码
class Solution {
public:
    vector<bool> isArraySpecial(vector<int>& nums, vector<vector<int>>& queries) {
        vector<int> group(nums.size());
        for(int i = 0;i<nums.size();i++){
            group[i] = (nums[i] % 2 == 0) ? 0 : 1;
        }
        vector<int> prefix(nums.size(), 0);
        for(int i = 1;i < nums.size();i++){
            prefix[i] = prefix[i-1];
            if(group[i] == group[i-1]){
                prefix[i]++;
            }
        }

        vector<bool> answer;
        for(const auto& query : queries){
            int sameCount = prefix[query[1]] - prefix[query[0]];
            answer.push_back(sameCount == 0);
        }
        return answer;
    }
};

所以为了避免一直重复的遍历数组,定义一个prefix用来储存group在 i 次前相同字符出现次数(也就是有两个false或者两个true相邻的次数)。所以只需要看在某个范围内,prefix[上限] 减去 prefix[下限]如果为0,说明没有出现过两个false和两个true相邻的情况,说明是特殊字符。

相关推荐
passer__jw76714 分钟前
【LeetCode】【算法】283. 移动零
数据结构·算法·leetcode
Ocean☾21 分钟前
前端基础-html-注册界面
前端·算法·html
顶呱呱程序29 分钟前
2-143 基于matlab-GUI的脉冲响应不变法实现音频滤波功能
算法·matlab·音视频·matlab-gui·音频滤波·脉冲响应不变法
爱吃生蚝的于勒1 小时前
深入学习指针(5)!!!!!!!!!!!!!!!
c语言·开发语言·数据结构·学习·计算机网络·算法
羊小猪~~1 小时前
数据结构C语言描述2(图文结合)--有头单链表,无头单链表(两种方法),链表反转、有序链表构建、排序等操作,考研可看
c语言·数据结构·c++·考研·算法·链表·visual studio
王哈哈^_^1 小时前
【数据集】【YOLO】【VOC】目标检测数据集,查找数据集,yolo目标检测算法详细实战训练步骤!
人工智能·深度学习·算法·yolo·目标检测·计算机视觉·pyqt
星沁城1 小时前
240. 搜索二维矩阵 II
java·线性代数·算法·leetcode·矩阵
脉牛杂德2 小时前
多项式加法——C语言
数据结构·c++·算法
legend_jz2 小时前
STL--哈希
c++·算法·哈希算法
kingmax542120082 小时前
初三数学,最优解问题
算法