improved to solve for any number of floors / given number of light bulbs