ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

1664. Ways to Make a Fair Array

2021-04-09 19:36:31  阅读:185  来源: 互联网

标签:even right Ways Make 元素 int 1664 odd left


思路:
这道题需要找规律了,因为题目是求有多少种方案获得平衡数组,那么就需要遍历所有元素。
首先 对于删除元素的左侧是可以发现不受影响的,但删除元素的右侧元素下标会-1,整体向左侧移动一个单位,那么原来在奇数位置的元素就会变到偶数位。所以我们只需要得到右侧的元素的和即可。
我们定义

int left_odd=0,right_odd=0;
int left_even=0,right_even=0;
int sum_odd=0,sum_even=0;

首先我们计算数组的奇数和 与 偶数和,然后遍历数组,从第一个元素开始,先判断是否处于第一个元素,如果是那么就不更新left_odd和left_odd,因为第一个元素左侧没有元素。如果不是就判断i-1的位置是偶数位还是奇数位,偶数位就给left_even相加,奇数就给left_odd加。这里也有一个知识点,这样积累前面数的和的为前缀和,也是一种算法。

我们已经知道删除元素右侧的奇偶位会交换,那么可以得到 一个结论,sum_odd-left_odd = right_even,因为在删除这个元素之前,等式左侧会得到右侧元素的奇数和,当删除元素后,下标向左移动,奇数和就变成了偶数和。同样 sum_even-left_even = right_odd。

接着在判断删除元素的位置是偶数还是奇数位,如果是偶数位,那么就应该right_odd减这个元素,因为这个元素本来就是偶数和里面的,然后right_odd是由总的偶数和减去左侧偶数和得到的,所以这个元素将会存在与right_odd中。
对于奇数位也是相同的。
最后在判断left_odd+right_odd 是否等于 left_even+right_even,相等就计数一次。

代码:

class Solution {
public:
    int waysToMakeFair(vector<int>& nums) {
        int n=nums.size();
        int left_odd=0,right_odd=0;
        int left_even=0,right_even=0;
        int sum_odd=0,sum_even=0;
        int cnt=0;
        for(int i=0;i<n;++i){
            if(i%2==0) sum_even+=nums[i];
            else sum_odd += nums[i];
        }
        for(int i=0;i<n;++i){
            if(i!=0){
                if((i-1)%2==0) left_even+=nums[i-1];
                else left_odd+=nums[i-1];
            }
            right_even = sum_odd - left_odd;
            right_odd = sum_even -left_even;
            if(i%2==0){
                right_odd-=nums[i];
            }
            else{
                right_even-=nums[i];
            }
            if(right_even+left_even==left_odd+right_odd) cnt++;
        }
        return cnt;
    }
};

标签:even,right,Ways,Make,元素,int,1664,odd,left
来源: https://www.cnblogs.com/Mrsdwang/p/14638494.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有