题目
给定一个由 0 和 1 组成的数组 A，将数组分成 3 个非空的部分，使得所有这些部分表示相同的二进制值。
如果可以做到，请返回任何 [i, j]，其中 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, -1]。
注意，在考虑每个部分所表示的二进制时，应当将其看作一个整体。例如，[1,1,0] 表示十进制中的 6，而不会是 3。此外，前导零也是被允许的，所以 [0,1,1] 和 [1,1] 表示相同的值。
示例 1：
输入：[1,0,1,0,1]
输出：[0,3]
示例 2：
输出：[1,1,0,1,1]
输出：[-1,-1]
提示：
3 <= A.length <= 30000
A[i] == 0 或 A[i] == 1
思路
有几个不太容易一眼就发现的小tips
- 分为三等份的相同值，其 ‘1’ 的个数应该是相同的，也就是说，在给定的数组中，’1‘ 的个数应该可以 % 3 == 0，否则无法分为三等份。可以分为三等份的每一份中’1‘的个数应该是数组中’1‘的个数除三。
- 知道‘1’的个数以后，我们可以确定下来每一份的值，该值是唯一的，可以通过最后一份来求出：从后往前遍历数组，直到‘1’的个数等于数组中‘1’的个数除三，该二进制数就是每一份的值。（该过程可以保留后缀的‘0’的数量以备后面用到）
- 确定值后，可以从前向后遍历数组，同样通过‘1’的个数找数，‘1’的个数找齐后，加上刚才求出后缀0，如果后面的0的数量没有求出的后缀0多，则无法分割。若有，求这个二进制数的值，等于之前求出的值，则这个位置就是i 的值，若不像等，则无法分割。重复该步骤可以得到j 的值。
该方法的时间复杂度为O(n)，如果大家有更优秀的解决办法，或者可以优化的地方，欢迎留言。
code
func threeEqualParts(A []int) []int {
var lA = len(A) -1
result := make([]int,0,2)
ones := 0
for _, a := range A {
if a == 1 {
ones++
}
}
if ones % 3 != 0{
result = append(result, -1, -1)
return result
}
partOnes := ones / 3
if partOnes == 0 {
result = append(result, 0, 4)
return result
}
tPartOnes := partOnes
suffixZero := 0
flag := false
realValue := 0
var tmpI, tmpJ int
for i:=lA;i>0;i-- {
// 确定真值 suffixZero
if !flag && A[i] == 0{
suffixZero++
}
if A[i] == 1 {
flag = true
tPartOnes--
if tPartOnes == 0{
realValue = cal(A[i:])
tmpJ = i
break
}
}
}
tmpI,ok := find(A[:tmpJ], partOnes, realValue, suffixZero)
if ! ok {
result = append(result, -1, -1)
return result
}
tmpJ, ok = find(A[tmpI+1:tmpJ], partOnes, realValue, suffixZero)
if ! ok {
result = append(result, -1, -1)
return result
}
tmpJ = tmpI+tmpJ+2
result = append(result, tmpI, tmpJ)
return result
}
// 用来找 1、2 部分，从前找，找到就返回当前位置
func find(A []int, partOnes, realValue, suffixZero int) (i int, b bool) {
l := len(A) -1
tmpI:=0
for i, a := range A {
if a == 1 {
partOnes--
if partOnes==0{
for j:=suffixZero;j>0;j--{
if i+j > l {
return 0, false
}
if A[i+j] != 0{
return 0, false
}
}
tmpI = i+suffixZero
break
}
}
}
if cal(A[:tmpI+1]) != realValue {
return 0, false
}
return tmpI, true
}
func cal(A []int) int {
l := len(A)
result := 0
for i, a := range A {
result += a * ((l-i) * (l-i))
}
return result
}