Distribute N candies among K children

candies_children.rb

candies = 10
no_of_children = 3
children = Array.new(no_of_children, 0) # [0,0,0, ...n]
main_index = 0

while(candies > 0)
  if candies > main_index
    children[main_index % no_of_children] += main_index + 1
    candies = candies - (main_index + 1)
  else
    children[main_index % no_of_children] += candies
    candies = 0
  end
  main_index = main_index + 1
end

children