我需要检查 Map 中是否已经存在添加值并将值添加到 Map 中的整数列表

Question

我需要创建一个以字符串为键、整数列表为值的 Map。我创建了地图，创建了从 txt 文件中获取的单词的标记化。现在我正在尝试将键和值放入映射中，但我仍然收到错误。我想知道也许我的地图创建错误，但我没有看到任何其他方式来创建此地图。

我一直在尝试用“put”函数添加这个地图，这样我什至无法编译这个程序

Map<String, List<Integer>> map = new HashMap<String, List<Integer>>();
        //Map<String, Integer> map = new HashMap<String, Integer>();    
        int numberOfFiles;
        File directory = new File("C:/Users/User/Desktop/documents java project/");
        numberOfFiles = directory.list().length;


        for(int i = 1; i <= numberOfFiles; i++) {
            //tokenization of string
            String filePath = "C:/Users/User/Desktop/documents java project/document "+ i +".txt";
            String fileData = readFile(filePath);
            String delimiter = " ";
            String fileWords [] = fileData.split(delimiter);
            //words.add(fileWords);
            int numberOfWords = fileWords.length;

            for(int j = 0; j < numberOfWords; j++) {

                map.put(fileWords[j], i);

            }
        }
}

Answer 1

您的Map 期望值是List<Integer>，但您正试图将int 放入其中：

map.put(fileWords[j], i);

相反，您需要创建一个List<Integer> 并将其作为值，例如：

if (!map.containsKey(fileWords[j])) {
    map.put(fileWords[j], new ArrayList<>());
}
map.get(fileWords[j]).add(i);

这会将新的空List 添加到您的Map（如果它尚不存在），然后它将i 的值添加到列表中。如果它已经存在，它只会将i 添加到已经存在的列表中。

如果以上内容对您的口味来说太罗嗦，也可以使用Map.computeIfAbsent() 来完成：

map.computeIfAbsent(fileWords[j], k -> new ArrayList<>()).add(i);

如果您不想存储重复项，可以改用Set。将您的声明更改为：

Map<String, Set<Integer>> map = new HashMap<>();

然后，添加项目：

map.computeIfAbsent(fileWords[j], k -> new HashSet<>()).add(i);