使用递归删除树形结构的所有子节点(java和mysql实现)

使用递归删除树形结构的所有子节点(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 t=new HashMap();

// 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 keys=t.keySet();

Iterator 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("—— —— —— —— —— ——删除掉的节点 —— ——http:// —— —— —— —— ——");

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));

}

}

//http://递归删除所有的子节点

public static Map removeTreeNodes(Map t,Integer k){

//所有需要删除的子节点

List sons=new ArrayList();

sons.add(k);

List temp=new ArrayList();

//循环递归删除，所有以k为父节点的节点

while(true){

for(Integer s:sons){

Set keys=t.keySet();

Iterator 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();

}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;

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。