IMHO you did very well if you were able to write down the correct naive solution and then discuss optimizations afterwards. Hopefully it works out for you! :)

Godspeed!