每日一题｜2022-10-6｜927. 三等分

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。