AEA Investors interview question

data structure used to store the data