【数组相关】1013. 将数组分成和相等的三个部分

原题链接:https://leetcode-cn.com/problems/partition-array-into-three-parts-with-equal-sum/



反证法,就假设当前数组可以被分为3部分,所以先计算总和 sum ,在计算每个部分的值 key=sum/3

然后数组从左到右遍历,每遍历一个元素,则让key减去它,直到key减为0,计数出现一组 group=1,key再重新赋值sum/3,接着再来,最终判断group是否为3即可

class Solution {
    public boolean canThreePartsEqualSum(int[] A) {
        int sum=0;
        for(int n:A){
            sum+=n;
        }
        int key=sum/3;
        int group=0;
        for(int i=0;i<A.length;i++){
            key-=A[i];
            if(key==0){
                group++;
                key=sum/3;
            }
        }
        return group==3;
    }
}
发布了88 篇原创文章 · 获赞 142 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/u010189239/article/details/104419165
今日推荐