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

时间:2019-02-21 02:50:12

标签: java

我正在用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还是其他?

任何建议,我将不胜感激。

2 个答案:

答案 0 :(得分: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
    }
}

答案 1 :(得分:1)

在这种情况下,

TreeSet 优于 HashMap ,因为它不接受重复的值,因此保证了更高的完整性。

无论如何, HashMap 更适合您的需求,因为 Path 可以作为键,而 word 可以作为值。

我准备了一个示例,说明如何检索 path word ,然后将它们放入 key value 分别为HashMap:

Map<String, String> fileParameters = new HashMap<>();
ResultSet rs = stmt.executeQuery("SELECT path, word FROM files");
while(rs.next()) {
String path = rs.getString("path");
String name = rs.getString("word");
fileParameters.put(path, name);