Coding: Infectious Disease Simulation | Dataford Interview Questions

Coding question about infectious diseases with multiple parts: 1) Only infection status, how long has it been infected? 2) Adding immunity status. 3) After D days of infection, how long until stability? 4) Including death status.

Implement analyze_disease(events, D, last_day). Each event is [person_id, day, status], where status is S, I, T, M, or X. Return an object keyed by person ID containing infected, infection_duration, immune, days_to_stability, and dead. Infection duration ends at the first T, M, or X event, or at the end of last_day if no ending event exists.

Constraints

1 <= events.length <= 10^5
Each event has the form [person_id, day, status]
0 <= person_id <= 10^9
0 <= day <= last_day
1 <= D <= last_day + 1
status is one of "S", "I", "T", "M", or "X"
Each person has at most one initial infection event

Function Signature

def analyze_disease(events, D, last_day):

Your question is Coding: Infectious Disease Simulation. Start with the requirements in the Question tab.

Run and submit as often as you like. When you're ready, talk me through your approach or go straight to the code.

Problem

Constraints

1 <= events.length <= 10^5
Each event has the form [person_id, day, status]
0 <= person_id <= 10^9
0 <= day <= last_day
1 <= D <= last_day + 1
status is one of "S", "I", "T", "M", or "X"
Each person has at most one initial infection event

Function Signature

def analyze_disease(events, D, last_day):

Your question is Coding: Infectious Disease Simulation. Start with the requirements in the Question tab.

Run and submit as often as you like. When you're ready, talk me through your approach or go straight to the code.