updated solution (including optimization) for problem 1 in harder problems. this was so much fun! :)