Pergunta de entrevista da empresa RAND

Please complete the following programming challenge. We aren't looking for exact programming syntax so you can write however you want (SAS, R, Python, etc)