927. 三等分
思路:计数+三指针
我们记数组的长度为n,数组中1的数量为cnt。
显然cnt必须是3的倍数,否则无法将数组三等分,则可以提前返回[-1,-1]。如果cnt为0,那么意味着数组中所有的元素都为0,直接返回[0, n-1]即可/或者直接返回[0,2],因为题目没要求说全是0的时候怎么分,所以我们只要满足条件的进行分就行啦。
当cnt是3的倍数的时候,我们将cnt除以3,得到每一份中1的数量,然后找到每一份的第一个1在数组arr中的位置(参考一下代码中find(x)函数),分别记为i,j,k。
0 1 1 0 0 0 1 1 0 0 0 0 0 1 1 0 0
^ ^ ^
i j k
接着我们从i,j,k开始往后同时遍历每一部分,判断三部分对应的值是否相等,是就继续遍历,直到k到达数组arr末尾。
0 1 1 0 0 0 1 1 0 0 0 0 0 1 1 0 0
^ ^ ^
i j k
遍历结束时,若k=n,则说明满足三等分,返回此时的[i-1, j] 作为答案,否则返回[-1, -1] 。
时间复杂度为O(n),空间复杂度为O(1),其中n表示arr的长度。
func threeEqualParts(arr []int) []int {
find := func(x int) int {
s := 0
for i, v := range arr {
s += v
if s == x {
return i
}
}
return 0
}
n := len(arr)
cnt := 0
for _, v := range arr { // cnt记录1的数量
cnt += v
}
if cnt%3 != 0 { // 如果1的个数不是3的倍数,肯定不能等分
return []int{-1,-1}
}
if cnt == 0 { // 如果全是0没有1,那么我们自行分割0,也可以返回[0, 2]
return []int{0, n-1}
}
cnt /= 3 // 1的个数分为三份
i, j, k := find(1), find(cnt+1), find(cnt*2+1) // 寻找三部分的1的起始位置
for ;k<n && arr[i] == arr[j] && arr[j] == arr[k];i,j,k = i+1,j+1,k+1 { // 三指针同时往后遍历判断
}
if k == n { // 如果遍历到最后都相等,说明可以三等分
return []int{i-1, j}
}
return []int{-1, -1} // 不能三等分
}
好好好,赚一道hard。