描述
We distribute some number of candies, to a row of n = num_people people in the following way:
We then give 1 candy to the first person, 2 candies to the second person, and so on until we give n candies to the last person.
Then, we go back to the start of the row, giving n + 1 candies to the first person, n + 2 candies to the second person, and so on until we give 2 * n candies to the last person.
This process repeats (with us giving one more candy each time, and moving to the start of the row after we reach the end) until we run out of candies. The last person will receive all of our remaining candies (not necessarily one more than the previous gift).
Return an array (of length num_people and sum candies) that represents the final distribution of candies.
Example 1:
Input: candies = 7, num_people = 4
Output: [1,2,3,1]
Explanation:
On the first turn, ans[0] += 1, and the array is [1,0,0,0].
On the second turn, ans[1] += 2, and the array is [1,2,0,0].
On the third turn, ans[2] += 3, and the array is [1,2,3,0].
On the fourth turn, ans[3] += 1 (because there is only one candy left), and the final array is [1,2,3,1].
Example 2:
Input: candies = 10, num_people = 3
Output: [5,2,3]
Explanation:
On the first turn, ans[0] += 1, and the array is [1,0,0].
On the second turn, ans[1] += 2, and the array is [1,2,0].
On the third turn, ans[2] += 3, and the array is [1,2,3].
On the fourth turn, ans[0] += 4, and the final array is [5,2,3].
Constraints:
1 <= candies <= 10^9
1 <= num_people <= 1000
解析
根据题意，其实就是把糖果按照 1、2、3、...n 的顺序依次发放给每个人，从头到尾周而复始，知道所有的糖果豆发完，值得注意的是最后一次发的糖果数，是剩余的糖果数。时间复杂度为 O(N)，空间复杂度为 O(N)。
解答
class Solution(object):
def distributeCandies(self, candies, num_people):
"""
:type candies: int
:type num_people: int
:rtype: List[int]
"""
result = [0]*num_people
// 预先给第一个人发糖
result[0]=1
// 从第二个人开始的发糖数
c = 2
// 糖果剩余数量
candies -= 1
// 数组索引
idx = 1
while candies >= c:
result[idx%num_people] += c
candies -= c
idx+=1
c+=1
if candies:
result[idx%num_people] += candies
return result
运行结果
Runtime: 16 ms, faster than 94.84% of Python online submissions for Distribute Candies to People.
Memory Usage: 11.9 MB, less than 100.00% of Python online submissions for Distribute Candies to People.
每日格言：生命不等于是呼吸，生命是活动。
请作者喝吃泡饭 支付宝
