Leetcode Note: Go - Distribute Candies to People
Distribute Candies to People - LeetCode
https://leetcode.com/problems/distribute-candies-to-people/
- Go 言語で取り組んだメモ
回答
Golang Beginner Solution - LeetCode Discuss
https://leetcode.com/problems/distribute-candies-to-people/discuss/361961/Golang-Beginner-Solution
func distributeCandies(candies int, num_people int) []int {
result := make([]int, num_people)
pidx := 0
ctr := 1
for candies > 0 {
// fmt.Println(pidx, candies, ctr)
if pidx == num_people {
pidx = 0
}
if candies < ctr {
result[pidx] += candies
break
}
result[pidx] += ctr
candies -= ctr
ctr++
pidx++
}
return result
}