一个简单的多叉树实现

import java.util.List;
import java.util.concurrent.CopyOnWriteArrayList;

public class Tree {
	
	Node node = null;
	
	public static void main(String[] args) {
		Tree tree = new Tree();
		tree.insert(0,1);
		tree.insert(1,2);
		tree.insert(1,3);
		tree.insert(3,4);
		tree.insert(3,5);
		
		tree.print(tree.node);
		System.out.println();
		
		tree.delete(3);
		tree.print(tree.node);
		System.out.println();
		
		tree.delete(2);
		tree.print(tree.node);
		System.out.println();
		
		tree.insert(1, 2);
		tree.print(tree.node);
		System.out.println();
		
		tree.insert(4, 6);
		tree.print(tree.node);
	}
	
	private void delete(int id) {
		if(node==null) {
			return;
		}
		
		remove(null,node,id);
	}

	private void remove(Node pNode,Node node, int id) {
		if(node.id==id && pNode!=null) {
			if(node.childrens!=null && node.childrens.size()>0) {
				for(Node childNode : node.childrens) {
					childNode.pid=node.pid;
					pNode.childrens.add(childNode);
				}
				
				for(Node pchildren:pNode.childrens) {
					if(pchildren.id==id) {
						pNode.childrens.remove(pchildren);
						break;
					}
				}
			}else {
				pNode.childrens.remove(node);
			}
		}else {
			for(Node childNode : node.childrens) {
				remove(node,childNode,id);
			}
		}
	}

	public synchronized void insert(int pid,int id) {
		if(node==null) {
			node = new Node();
			addFirst(node,pid,id);
		}else {
			addNode(node,pid,id);
		}
	}
	
	private void addNode(Node node, int pid, int id) {
		if(node.id==pid) {
			Node newNode = new Node();
			newNode.id=id;
			newNode.pid=pid;
			node.childrens.add(newNode);
		}else {
			for(Node childNode : node.childrens) {
				addNode(childNode,pid,id);
			}
		}
	}

	private void addFirst(Node node,int pid, int id) {
		node.id=id;
		node.pid=pid;
	}
	
	public void print(Node node) {
		if(node.childrens != null && node.childrens.size() > 0) {
			for(Node childNode : node.childrens) {
				print(childNode);
			}
		}
		System.out.println(node.id+":"+node.pid);
	}

	static class Node{
		public int id;
		public int pid;
		public List<Node> childrens = new CopyOnWriteArrayList<>();
	}
}

输出:

2:1

4:3

5:3

3:1

1:0

 

2:1

4:1

5:1

1:0

 

4:1

5:1

1:0

 

4:1

5:1

2:1

1:0

 

6:4

4:1

5:1

2:1

1:0

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值