Use Set[] instead of HashMap<Integer, Set>