Simulate Plant Infection Spread | OpenAI Interview Question