LeetCode刷题笔记——三指针模拟（927. 三等分）

LeetCode刷题笔记——三指针模拟（927. 三等分）

927. 三等分 困难

给定一个由 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：

输入：arr = [1,0,1,0,1]
输出：[0,3]

示例 2：

输入：arr = [1,1,0,1,1]
输出：[-1,-1]

示例 3:

输入：arr = [1,1,0,0,1]
输出：[0,2]

提示：

• 3 <= arr.length <= 3 * 10^4
• arr[i] 是 0 或 1

题解

• 如果1的数目s不是3的整数倍一定不能三等分。

• 令k=s/3如果k==0即全为零的数组，返回{0,2}或者其他满足条件的解。

• 指针a,b,c初始化时分别定位第1,k+1,2k+1个1.

• 如果扫描过程中三指针的值完全匹配（表示的二进制一样），c可到达n，即返回答案{a-1,b}.

• 如果扫描过程中三指针的值不匹配，c无法到达n，返回{-1,-1}.

Java

class Solution {
    public int[] threeEqualParts(int[] arr) {
        int n = arr.length, s = Arrays.stream(arr).sum(), k = s/3;
        if (s%3 != 0) return new int[]{-1,-1};
        if (k == 0) return new int[]{0,2};
        int a = find(arr,1),b = find(arr,k+1),c = find(arr,2*k+1);
        while (c < n && arr[a] == arr[b] && arr[b] == arr[c]) {a++;b++;c++;}
        if (c == n) return new int[]{a-1,b};
        return new int[]{-1,-1};
    }
    int find(int[] arr,int x) {
        for (int i = 0; i < arr.length; i++) {
            x -= arr[i];
            if (x == 0) return i;
        }
        return -1;
    }
}

Go

func threeEqualParts(arr []int) []int {
    n,s := len(arr),0
    for _,v := range arr {s += v}
    if s%3 != 0 {return []int{-1,-1}}
    k := s/3
    if k == 0 {return []int{0,2}}
    find := func(x int) int {
        for i := 0; i < n; i++ {x -= arr[i];if x == 0 {return i;}}
        return -1
    }
    a,b,c := find(1),find(k+1),find(2*k+1)
    for c < n && arr[a] == arr[b] && arr[b] == arr[c] {a++;b++;c++;}
    if c == n {return []int{a-1,b}}
    return []int{-1,-1}
}