Compute time to infect all cells | OpenAI Interview Question