LeetCode练习DAY1

LeetCode 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] 表示相同的值。

class Solution(object):
    def threeEqualParts(self, arr):
        def find(x):#寻找第i个1 的下标
            s=0
            for i,v in enumerate(arr):
                s+=v
                if s==x:
                    return i

        n=sum(arr)
        if n%3:#不能平等的分成三分，不可能相等
            return [-1,-1]
        if n==0:#都是0，随便分
            return [0,2]    
        
        #能分成三份，找到first，second,和third三个下标的位置
        m=n/3
        i, j, k = find(1), find(m + 1), find(m* 2 + 1)
        print (i,j,k)
        while k < len(arr) and arr[i] == arr[j] == arr[k]:
            i, j, k = i + 1, j + 1, k + 1
        if k==len(arr):
            return [i-1,j]
        else:
            return [-1,-1]

        

        """
        解题思路：
        1、首先我们将字符串中的1分成三等份：如果不能平均分，则说明无法做到，否则将按照2、的情况进行讨论
        2、分别找到第一组，第二组，第三组中第一个1的位置，从第三组可以知道有效二进制字符串长度为L=len(arr)-third,对三个组的长度为L的后缀字符比较，如果相同，则最终的结果为[first+L-1,second+L]
        :type arr: List[int]
        :rtype: List[int]
        """