Simulate Grid Infection | OpenAI Interview Question