我正在用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还是其他?
任何建议,我将不胜感激。
答案 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);