Google DeepMind interview question

Solve a hackerrank-esque python problem in a Jupyter notebook