場景
- 背景
數據庫中有一張存儲樹結構的表,表中有子節點(childId)和父節點(parentId)對應關系的字段。表中存儲多棵樹的數據,每棵樹的根節點(rootId)的特點是:childId與parentId相等。數據如下圖:
Paste_Image.png
- 需求
表中每棵樹都可能有多級分支及多個葉子節點,需要定位出表中所有節點歸屬的樹的根節點Id,放入map中(此處為java實現),map的結構為map<childId,rootId>。最終結果如下圖:
Paste_Image.png
思路
將表數據拼接取出,然后將數據轉換為兩個數組,一個是父節點數組、一個是子節點數組,同時,提取出表中的根節點數據。
然后,對子節點數組進行遍歷,依據已經確定的根節點,遞歸查找出每個子節點的根,當子節點數據遍歷完成,則所有數據定位完成。
實現代碼
從數據庫獲取表數據,拼接成“childId,parentId”字符串。可以通過jdbc從查詢表,獲取resultSet,遍歷resultSet進行拼接,比較簡單,所以未編寫代碼。
/**
* 多樹表,遞歸定位所有分支及葉子的根
* @param child_parent_list
* @author 張文健
* @return
*/
public static Map<String,String> rootMap(List<String> child_parent_list){
Map<String,String> child_parent_map = new HashMap<String,String>();
String[] parentArr = new String[child_parent_list.size()];
String[] childArr = new String[child_parent_list.size()];
for(int i=0;i<child_parent_list.size();i++){
String ic = child_parent_list.get(i).split(",")[0];
String ip = child_parent_list.get(i).split(",")[1];
parentArr[i] = ip;
childArr[i] = ic;
if(ic.equals(ip)){
child_parent_map.put(ip, ip);
}
}
goods: for(int i=0;i<childArr.length;i++){
String finder = parentArr[i];
machine: for(int j=0;j<childArr.length;j++){
if(finder.equals(childArr[j])){
if(child_parent_map.containsValue(childArr[j])){
child_parent_map.put(childArr[i], childArr[j]);
continue goods;
}else{
finder = parentArr[j];
j=0;
continue machine;
}
}
}
}
return child_parent_map;
}
public static void main(String[] args) {
List<String> child_parent_list = new ArrayList<String>();
child_parent_list.add("000087,000086");
child_parent_list.add("000088,000086");
child_parent_list.add("000105,000088");
child_parent_list.add("000106,000105");
child_parent_list.add("000086,000086");
child_parent_list.add("000091,000091");
child_parent_list.add("000092,000091");
child_parent_list.add("000093,000091");
child_parent_list.add("000096,000096");
child_parent_list.add("000097,000096");
child_parent_list.add("000098,000096");
child_parent_list.add("000099,000096");
child_parent_list.add("000113,000111");
child_parent_list.add("000112,000111");
child_parent_list.add("000114,000111");
child_parent_list.add("000111,000111");
ServiceUtil.rootMap(child_parent_list);
}