(nice!!!)LeetCode 3152. 特殊数组 II（前缀和）

岁忧

于 2024-08-14 08:56:18 发布

阅读量97

点赞数 4

分类专栏： LeetCode 文章标签： leetcode 算法职场和发展 c++ 数据结构前缀和

本文链接：https://blog.csdn.net/weixin_46028214/article/details/141180652

版权

LeetCode 专栏收录该内容

418 篇文章 0 订阅

订阅专栏

题目：3152. 特殊数组 II

在这里插入图片描述

思路：如果某个元素nums[i]与其前一个元素nums[i-1]奇偶性一致，那么标记当前sum[i]值为1；反之，则为0。
然后求出sum的前缀和。如果区间[x,y]为特殊数组，则sum[x]==sum[y]

class Solution {
public:
    vector<bool> isArraySpecial(vector<int>& nums, vector<vector<int>>& queries) {
        int n=nums.size();
        vector<int> sum(n,0);
        for(int i=1;i<n;i++){
            if(nums[i]%2==nums[i-1]%2){
                sum[i]=sum[i-1]+1;
            }else{
                sum[i]=sum[i-1];
            }
        }
        vector<bool> v;
        for(int i=0;i<queries.size();i++){
            int x=queries[i][0],y=queries[i][1];
            if(sum[x]==sum[y]) v.push_back(true);
            else v.push_back(false);
        }
        return v;
    }
};