OpenAI interview question

Graphs question (Human Virus Infection problem)