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]
        """

你可能感兴趣的:(python)