java-如何将单词和文件提取到一个集合中？

Question

我正在用 Java 创建一个简单的算法，该算法可以从数据库表中提取单词及其各自的文件，并将其保存在集合或数组中。

例如，我的表格如下所示:

path        word
file1        w1

file1        w2


file1        w3

.........

file2        w2

file2        w5

.........

这样的例子不胜枚举。

在我的程序中，我想从表中提取这些数据，以便将其存储在如下所示的集合中:

w1={file1}
w2={file1, file2}
w3={file1}
w5={file2}
...... and etc

当然，表中肯定会有更多数据，但这只是我想要完成的总体思路。

我要做的第一步是建立与数据库的 JDBC 连接并从表中运行 select 语句。然而，我不知道如何以一种像我上面描述的那样存储它们的方式提取它们。

我应该使用数组、hashSet 还是其他东西？

如果有任何建议，我将不胜感激。

Answer 1

您可以使用HashMap<String,TreeSet<String>> map = new HashMap<>();

那么你的代码将是:

ResultSet rs = stmt.executeQuery("SELECT PATH, WORD FROM TABLE_A");

while(rs.next()) {
    if (map.containsKey(rs.getString("WORD"))) { // If the word is already in your hash map
        TreeSet<String> path = map.get(rs.getString("WORD")); //get the set of files where this word exist 
        path.add(rs.getString("PATH")); // add the new path to the set
        map.put(rs.getString("WORD"), path); // update the map
    } else { // else if the word is new
        TreeSet<String> path = new TreeSet<String>(); // create a new set
        path.add(rs.getString("PATH")); // add the path to the set
        map.put(rs.getString("WORD"), path); // add the new data to the map
    }
}