使用遞歸怎么刪除樹形結構的所有子節點

使用遞歸怎么刪除樹形結構的所有子節點？很多新手對此不是很清楚，為了幫助大家解決這個難題，下面小編將為大家詳細講解，有這方面需求的人可以來學習下，希望你能有所收獲。

1.業務場景

有如下樹形結構:
+—0
+—1
+—2
+—4
+—5
+—3

如果刪除某個父節點，則其子節點，以及其子節點的子節點，以此類推，需要全部刪除。

2.Java實現

使用Map存儲樹形結構的數據，id為map的key，pid為樹形結構的value。

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.CopyOnWriteArrayList;

public class TreeNodes {
  public static void main(String[] args) {
    test();
  }

  //測試removeSons方法
  public static void test(){

    //原始的Map
    Map<Integer, Integer> t=new HashMap<Integer, Integer>();
    //  ID PID
    t.put(1, 0);
    t.put(2, 1);
    t.put(3, 1);
    t.put(4, 2);
    t.put(5, 4);
    System.out.println("—— —— —— —— —— ——原始的Map —— —— —— —— —— —— ——");

    Set<Integer> keys=t.keySet();
    Iterator<Integer> iterator=keys.iterator();
    System.out.println("ID —— PID");
    while(iterator.hasNext()){
      Integer i=iterator.next();
      System.out.println(i+" —— "+t.get(i));  
    }

    int k=2;
    //遞歸刪除k的所有子節點
    System.out.println("—— —— —— —— —— ——刪除掉的節點 —— —— —— —— —— —— ——");
    removeTreeNodes(t,k);
    //刪除k節點本身
    t.remove(k);
    System.out.println();
    System.out.println("—— —— —— —— —— —— 遞歸刪除["+k+"]的所有子節點后的Map —— —— ");

    iterator=keys.iterator();
    System.out.println("ID —— PID");
    while(iterator.hasNext()){
      Integer i=iterator.next();
      System.out.println(i+" —— "+t.get(i));
    }
  }

  //遞歸刪除所有的子節點
  public static Map<Integer, Integer> removeTreeNodes(Map<Integer, Integer> t,Integer k){ 
    //所有需要刪除的子節點
    List<Integer> sons=new ArrayList<Integer>();
    sons.add(k);
    List<Integer> temp=new ArrayList<Integer>();
    //循環遞歸刪除，所有以k為父節點的節點
    while(true){    
      for(Integer s:sons){
        Set<Integer> keys=t.keySet();
        Iterator<Integer> it=keys.iterator();
        while(it.hasNext()){
          Integer n=it.next();
          //如果父節點（即Map的value）為需要刪除的節點，則記錄此節點，并在Map中刪除
          if(t.get(n)==s){
            temp.add(n);
            it.remove();
            System.out.println("刪除了ID=["+n+"]的節點,其父節點為["+s+"]");
          }
        }
      }

      //如果此節點包含子節點，則將子節點賦值給sons;否則表示所有子節點已經刪除，結束循環
      if(temp.size()>0){
        sons=temp;  
        temp=new CopyOnWriteArrayList<Integer>();
      }else{
        break;
      }
    }

    return t;
  }
}

3.SQL實現

利用存儲過程，將所有子節點存儲到臨時表里。

存儲過程執行完后會產生一個臨時表，id為需要刪除的子節點id,nLevel 為節點深度，sCort 為排序字段。

建表并插入數據：

CREATE TABLE treeNodes
(
 id INT PRIMARY KEY,
 pid INT
);
INSERT INTO treeNodes VALUES 
(1, 0),
(2, 1),
(3, 1),
(4, 2),
(5, 4);

創建并調用存儲過程：

DELIMITER//

DROP PROCEDURE IF EXISTS removeTreeNodes//

CREATE PROCEDURE removeTreeNodes(IN rootid INT)
 BEGIN
 DECLARE LEVEL INT ;
 DROP TABLE IF EXISTS tempNodes;
 CREATE TABLE tempNodes (
  id INT,
  nLevel INT,
  sCort VARCHAR(8000)
 );
 SET LEVEL=0 ;
 INSERT INTO tempNodes SELECT id,LEVEL,ID FROM treeNodes WHERE PID=rootid;
 WHILE ROW_COUNT()>0 DO
  SET LEVEL=LEVEL+1 ;
  INSERT INTO tempNodes 
  SELECT A.ID,LEVEL,CONCAT(B.sCort,A.ID) FROM treeNodes A,tempNodes B 
  WHERE A.PID=B.ID AND B.nLevel=LEVEL-1 ;
 END WHILE;
 END;
//
DELIMITER ;

CALL removeTreeNodes(0);

下面是需要刪除的子節點：

SELECT CONCAT(SPACE(B.nLevel*2),'+--',A.id)
FROM treeNodes A,tempNodes B 
WHERE A.ID=B.ID 
ORDER BY B.sCort;

看完上述內容是否對您有幫助呢？如果還想對相關知識有進一步的了解或閱讀更多相關文章，請關注億速云行業資訊頻道，感謝您對億速云的支持。