LeetCode 927. 三等分

原题地址927. 三等分

---

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

---

算法 

就是把数组分成3部分，先保证每部分的1的个数是一样的，以及尾部的0的个数，再转换成整数形式判断值是否相等。

---

代码 

class Solution:
    def threeEqualParts(self, A):
        """
        :type A: List[int]
        :rtype: List[int]
        """
        sumA = sum(A)
        if sumA == 0:
            return [0, 2]
        if sumA % 3:
            return [-1, -1]
        cal = sumA // 3
        tail0, i = 0, len(A) - 1
        while A[i] == 0:
            i -= 1
            tail0 += 1
        lst = [[], [], []]
        tmp, idx, tail = 0, 0, 0
        for a in A:
            if tmp < cal:
                tmp += a
                lst[idx].append(str(a))
            elif tail < tail0:
                if a == 0:
                    lst[idx].append(str(a))
                    tail += 1
                else:
                    return [-1, -1]
            else:
                tmp, tail = 0, 0
                idx += 1
                tmp += a
                lst[idx].append(str(a))
        if int("".join(lst[0])) == int("".join(lst[1])) == int("".join(lst[2])):
            return [len(lst[0]) - 1, len(lst[0]) + len(lst[1])]
        else:
            return [-1, -1]