Orion Health Interview Question

What data structure would you use for this problem?