Python 每日一练leetcode

分糖果

排排坐，分糖果。

我们买了一些糖果 candies，打算把它们分给排好队的 n = num_people 个小朋友。

给第一个小朋友 1 颗糖果，第二个小朋友 2 颗，依此类推，直到给最后一个小朋友 n 颗糖果。

然后，我们再回到队伍的起点，给第一个小朋友 n + 1 颗糖果，第二个小朋友 n + 2 颗，依此类推，直到给最后一个小朋友 2 * n 颗糖果。

重复上述过程（每次都比上一次多给出一颗糖果，当到达队伍终点后再次从队伍起点开始），直到我们分完所有的糖果。注意，就算我们手中的剩下糖果数不够（不比前一次发出的糖果多），这些糖果也会全部发给当前的小朋友。

返回一个长度为 num_people、元素之和为 candies 的数组，以表示糖果的最终分发情况（即 ans[i] 表示第 i 个小朋友分到的糖果数）。

class Solution(object):
    def distributeCandies(self,candies,num_people):
        #定长度0数组的生成可用一句话表示
        ans=[0]*num_people
        i=0
        while candies!=0:
            #求余方法可以确定到每一位数组元素
            ans[i%num_people]+=min(i+1,candies)
            candies-=min(i+1,candies)
            i+=1
        return ans

三等分

给你一个整数数组 A，只有可以将其划分为三个和相等的非空部分时才返回 true，否则返回 false。

形式上，如果可以找出索引 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]) 就可以将数组三等分。

class Solution(object):
    def canThreePartsEqualSum(self, A):
        A_length=len(A)
        A_sum=sum(A)
        single=A_sum/3
        s=0
        l=[]
        for i in range(A_length):
            s+=A[i]
            if s==single:
                l.append(s)
                s=0
                if