Trying to find the right data structure to match the problem.