题目描述:
给定一个由 0 和 1 组成的数组 arr ,将数组分成 3 个非空的部分 ,使得所有这些部分表示相同的二进制值。
如果可以做到,请返回任何 [i, j],其中 i+1 < j,这样一来:
arr[0], arr[1], …, arr[i] 为第一部分;
arr[i + 1], arr[i + 2], …, arr[j - 1] 为第二部分;
arr[j], arr[j + 1], …, arr[arr.length - 1] 为第三部分。
这三个部分所表示的二进制值相等。
如果无法做到,就返回 [-1, -1]。
注意,在考虑每个部分所表示的二进制时,应当将其看作一个整体。例如,[1,1,0] 表示十进制中的 6,而不会是 3。此外,前导零也是被允许的,所以 [0,1,1] 和 [1,1] 表示相同的值。
题目思路:(参考官方了解题思路)
1.这个题目的主要思路就是如何把给定的数组分为三个部分,然后又如何去确定一个固定的区间的值,如果有不符合题目要求的直接输出[-1,-1];
2.第一个优化的点:统计数组中1的个数,如果不是3的倍数,不符合题目要求,直接返回[-1,-1];
3.如果1的个数为0的话,而且数组中最少有3个数,我们直接返回[0,2]就可以,当然也可以根据数组的长度自己定义;
4.然后我们要找到三个部分中,每一个1开始的位置,也就是0位置将1位置隔离开;
5.确定好三个部分后,因为最后一个部分到数组最后的位置是确定的,所以,我们反回来去确定第一个部分和第二个部分,通过循环,每次++,如果有不相等的,直接返回[-1,-1],否则,最后循环结束,返回符合要求的位置;
题目代码:
class Solution {
public int[] threeEqualParts(int[] arr) {
//利用Arrays.stream(arr)计算arr数组中的和---》也就是确定有多少个1
int sum = Arrays.stream(arr).sum();
//如果1的个数不能被整除
if (sum % 3 != 0) {
return new int[]{-1, -1};
}
//如果sum的和为0 直接返回
if (sum == 0) {
return new int[]{0, 2};
}
//将数组划分为三个部分
int partial = sum / 3;
//分别去确定三个区间的开始位置
int first = 0, second = 0, third = 0, cur = 0;
for (int i = 0; i < arr.length; i++) {
if (arr[i] == 1) {
if (cur == 0) {
first = i;
} else if (cur == partial) {
second = i;
} else if (cur == 2 * partial) {
third = i;
}
cur++;
}
}
//根据第三个区间位置的长度去确定第一个和第二个部分
int len = arr.length - third;
if (first + len <= second && second + len <= third) {
int i = 0;
while (third + i < arr.length) {
//如果不符合要求 直接结束
if (arr[first + i] != arr[second + i] || arr[first + i] != arr[third + i]) {
return new int[]{-1, -1};
}
//每次循环直接++
i++;
}
//返回最后的结果
return new int[]{first + len - 1, second + len};
}
return new int[]{-1, -1};
}
}