0模拟困难 LeetCode927. 三等分

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 * 104
arr[i] 是 0 或 1

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/three-equal-parts
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

分析

如果1的个数不是3的倍数则不能分割为3份。
容易错的点：
整个数组分为A,B,C三段，只有C段最后一个1后面的0是没有办法当作前导零的，所以只要A,B段包含C段第一个1到数组结尾的内容，那么一定可以分成三份，接下来就是要怎么分了。
首先要把A,B段剩余的0当作B,C的前导零，所以返回结果的第一个元素应该是A的起点加上C段的长度，第二个元素应该是B段的起点加上C段的长度。

以第三段的为基准，三段就是有1的部分是否一致(忽略前导零看三个数值是不是一致的)，然后把A,B剩余的0，当作B,C的前导零，去计算实际分割点。

class Solution {
    public int[] threeEqualParts(int[] arr) {
        int sum = 0;
        for (int num : arr) {
            sum += num;
        }
        if (sum % 3 != 0) {
            return new int[]{-1,-1};
        }
        if (sum == 0) {
            return new int[]{0,arr.length-1};
        }
        sum /= 3;
        int a = -1, b = -1, c = -1, tmp = 0;
        for (int i = 0; i < arr.length; i++) {
            tmp += arr[i];
            if (a == -1 && tmp == 1) a = i;
            if (b == -1 && tmp == sum + 1) b = i;
            if (c == -1 && tmp == 2 * sum + 1) c = i;
        }
        int n = arr.length - c;
        if (c - b < n || b - a < n) return new int[]{-1,-1};
        for (int i = 0; i < n; i++) {
            if (arr[a+i] != arr[b+i] || arr[b+i] != arr[c+i]) return new int[]{-1,-1};
        }
        return new int[]{a+n-1,b+n};
    }
}