来源于力扣 1013
题目要求: 给定一个整数数组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,如果sum不能够整除3,直接返回false。接下来看能不能分出来三部分,算出avg=sum/3,循环数组,如果累加的和等于avg,记下下标i,进行第二次循环;从i+1开始循环,如果累加的和等于avg,返回true;剩下的数之和肯定等于avg。
java代码
package 面试金典;
public class test015 {
public static void main(String[] args) {
int[] nums= {3,3,6,5,-2,2,5,1,-9,4};
boolean boo=PredictTheWinner(nums);
System.out.println(boo);
}
private static boolean PredictTheWinner(int[] A) {
int sum=0;
for(int i=0;i<A.length;i++) {
sum+=A[i];
}
if(sum%3==0) {
int avg=sum/3;
int avg1=0;
int avg2=0;
int s1=0;
for(int j=0;j<A.length;j++) {
avg1+=A[j];
if(avg1==avg) {
final int s=j;//final修饰的变量就变成了常量,即值不再变化
s1=s;
break;
}
}
for(int k=s1+1;k<A.length;k++) {
avg2+=A[k];
if(avg2==avg) {
return true;
}
}
}
return false;
}
}