【简单】Leetcode-将数组分成和相等的三个部分///细节处理///前置后置递增谨慎选择

给你一个整数数组 A,只有可以将其划分为三个和相等的非空部分时才返回 true,否则返回 false。
形式上,如果可以找出索引 i+1 < j 且满足 (A[0] + A[1] + … + A[i] == A[i+1] + A[i+2] + … + A[j-1] == A[j] + A[j-1] + … + A[A.length - 1]) 就可以将数组三等分。
示例 1:
输出:[0,2,1,-6,6,-7,9,1,2,0,1]
输出:true
解释:0 + 2 + 1 = -6 + 6 - 7 + 9 + 1 = 2 + 0 + 1

示例 2:
输入:[0,2,1,-6,6,7,9,-1,2,0,1]
输出:false

示例 3:
输入:[3,3,6,5,-2,2,5,1,-9,4]
输出:true
解释:3 + 3 = 6 = 5 - 2 + 2 + 5 + 1 - 9 + 4

思路:我刚开始的思路是先用sum计算累加和,如果能被3除尽 则除3然后继续,否则返回false。接下来用cur_sum记录每次累加一部分的和,当等于sum的时候,cnt加1。最后退出循环后判断,当cnt正好是3,并且cur_sum中没有别的数时,返回true,否则返回false。

bool canThreePartsEqualSum(vector<int>& A) {
        int sum = 0, cur_sum = 0, cnt = 0;
        for (int i = 0; i < A.size(); i++) sum += A[i];
        if (sum % 3 == 0) sum /= 3;
        else return false;
        for (int i = 0; i < A.size(); i++)
        {
            cur_sum += A[i];
            if (cur_sum == sum)
            {
                ++cnt;
                cur_sum = 0;
            }
        }
        if (cnt == 3 && cur_sum == 0) return true;
        return false;
    }

但是: 这个做法有一个样例无法通过:
输入:[10,-10,10,-10,10,-10,10,-10]
输出:false
预期:true
可以看出这个算法有一个缺陷,没考虑到sum=0的情况,正如这个样例所示,最后cnt算出来由4个,但其实可以[10,-10,10,-10] [10,-10] [10,-10]这样划分。所以最后的判断cnt应该大于等于3.

bool canThreePartsEqualSum(vector<int>& A) {
        int sum = 0, cur_sum = 0, cnt = 0;
        for (int i = 0; i < A.size(); i++) sum += A[i];
        if (sum % 3 == 0) sum /= 3;
        else return false;
        for (int i = 0; i < A.size(); i++)
        {
            cur_sum += A[i];
            if (cur_sum == sum)
            {
                ++cnt;
                cur_sum = 0;
            }
        }
        return cnt >= 3;
    }

2.双指针法
思路:参考了题解区一位老哥的做法。两边开始遍历,当两边是sum的时候,则中间没遍历到的一定也是sum。注意while循环条件,lef和right中间必须至少有一个,避免[1, -1, 1, -1]的情况发生。
accumulate详解

bool canThreePartsEqualSum(vector<int>& A) {
        int left = 0, right = A.size() - 1;
        int left_sum = A[left], right_sum = A[right];
        int sum = accumulate(A.begin(), A.end(), 0);
        if (sum % 3 != 0) return false;
        sum /= 3;
        while (right - left > 1) //right和left之间至少有1个
        {
            if (left_sum == sum && right_sum == sum) return true;
            if (left_sum != sum) left_sum += A[++left];
            if (right_sum != sum) right_sum += A[--right];
        }
        return false;
    }

注意写成这样会有样例无法通过:

class Solution {
public:
    bool canThreePartsEqualSum(vector<int>& A) {
        int left = 0, right = A.size() - 1;
        int left_sum = 0, right_sum = 0;  //区别
        int sum = accumulate(A.begin(), A.end(), 0);
        if (sum % 3 != 0) return false;
        else sum /= 3;
        while (right - left > 1)
        {
            if (left_sum == sum && right_sum == sum) return true;
            if (left_sum != sum) left_sum += A[left++];  //区别
            if (right_sum != sum) right_sum += A[right--]; //区别
        }
        return false;
    }
};

因为是后置递增,所以先累加之后再自增,而自增之后可能就无法通过while循环了,
比如样例:
输入:[3,3,6,5,-2,2,5,1,-9,4]
输出:false
预期:true
在right走到A[3]的5然后累加之后,right直接走到A[2]了,这时将不执行while循环内代码,所以不会返回true。直接返回false。
而前置递增先加之后直接递增,避免了这种情况。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值