循環(huán)暴力求解。
自增取余求people下標(biāo)
idx = (idx + 1)%num_people
對(duì)于最后一個(gè)被分糖果的people,需要把多余且不夠完全分的給加上
ret[idx] += candies + add
完整代碼
func distributeCandies(candies int, num_people int) []int {
add := 1
ret := make([]int, num_people)
idx := 0
candies -= add
for candies > 0 {
ret[idx] += add
add++
idx = (idx + 1)%num_people
candies -= add
}
ret[idx] += candies + add
return ret
}