原题链接: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;
}
}