递归删除树形结构的所有子节点（java和mysql实现）

---

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;