Can you provide the solution to the USACO Photoshoot Problem

Here is the link to my solution: https://repl.it/@DhruvYalamanchi/USACO-2020-January-Contest-Bronze-Problem-2

For USACO problems, there may very well be more than one correct method. This was how I approached it though.