Leetcode::3432. 统计元素和差值为偶数的分区方案

3432. 统计元素和差值为偶数的分区方案

已解答

简单

相关企业

提示

给你一个长度为 n 的整数数组 nums 。

分区是指将数组按照下标 i （0 <= i < n - 1）划分成两个非空子数组，其中：

左子数组包含区间 [0, i] 内的所有下标。
右子数组包含区间 [i + 1, n - 1] 内的所有下标。

对左子数组和右子数组先求元素和再做差，统计并返回差值为偶数的分区方案数。

示例 1：

**输入：**nums = [10,10,3,7,6]

**输出：**4

解释：

共有 4 个满足题意的分区方案：

[10]、[10, 3, 7, 6] 元素和的差值为 10 - 26 = -16 ，是偶数。
[10, 10]、[3, 7, 6] 元素和的差值为 20 - 16 = 4，是偶数。
[10, 10, 3]、[7, 6] 元素和的差值为 23 - 13 = 10，是偶数。
[10, 10, 3, 7]、[6] 元素和的差值为 30 - 6 = 24，是偶数。

示例 2：

**输入：**nums = [1,2,2]

**输出：**0

解释：

不存在元素和的差值为偶数的分区方案。

示例 3：

**输入：**nums = [2,4,6,8]

**输出：**3

解释：

所有分区方案都满足元素和的差值为偶数。

提示：

2 <= n == nums.length <= 100
1 <= nums[i] <= 100

暴力解法：

cpp 复制代码

class Solution {
public:
    int countPartitions(vector<int>& nums) {
        
        int res=0;
        int ans=0,arr=0;
        for(int i=1;i<nums.size();i++)
        {
            ans=0;
            arr=0;
            for(int j=0;j<i;j++){
                ans+=nums[j];
            }
            for(int j=i;j<nums.size();j++){
                arr+=nums[j];
            }

            if(abs(ans-arr)%2==0){
                res++;
            }
        }

        return res;
    }
};

前缀和:

cpp 复制代码

class Solution {
public:
    int countPartitions(vector<int>& nums) {

        vector<int>ans;
        int res=0,k=0;

        for(int i=0;i<nums.size();i++){
            k+=nums[i];
            ans.push_back(k);
        }     

        for(int i=1;i<nums.size();i++){
            int t=ans[nums.size()-1]-ans[i-1];
            if((t-ans[i-1])%2==0){
                res++;
            }
        }

        return res;
    }
};