代码展示
package demo5;
public class BinaryTree {
TreeNode root;
//设置根节点
public void setRoot(TreeNode root){
this.root = root;
}
//获取根节点
public TreeNode getRoot(){
return root;
}
public void frontShow() {
if(root!=null){
root.frontShow();
}
}
public void midShow() {
if(root!=null){
root.midShow();
}
}
public void afterShow() {
if(root!=null){
root.afterShow();
}
}
public TreeNode preSearch(int i) {
return root.preSearch(i);
}
public TreeNode midSearch(int i) {
return root.midSearch(i);
}
public TreeNode afterSearch(int i) {
return root.afterSearch(i);
}
public void delete(int i) {
if(root.value==i){
root = null;
}else{
root.delete(i);
}
}
}
package demo5;
public class TreeNode {
//节点的权
int value;
//左儿子
TreeNode leftNode;
//右儿子
TreeNode rightNode;
public TreeNode(int value){
this.value = value;
}
//设置左儿子
public void setLeftNode(TreeNode leftNode) {
this.leftNode = leftNode;
}
//设置右儿子
public void setRightNode(TreeNode rightNode) {
this.rightNode = rightNode;
}
//前序遍历
public void frontShow() {
//先遍历当前节点的内容
System.out.print(value+" ");
//左节点
if(leftNode!=null){
leftNode.frontShow();
}
//右节点
if(rightNode!=null){
rightNode.frontShow();
}
}
//中序遍历
public void midShow() {
//左节点
if(leftNode!=null){
leftNode.midShow();
}
//当前节点
System.out.print(value+" ");
//右节点
if(rightNode!=null){
rightNode.midShow();
}
}
//后序遍历
public void afterShow() {
//左节点
if(leftNode!=null){
leftNode.afterShow();
}
//右节点
if(rightNode!=null){
rightNode.afterShow();
}
//当前节点
System.out.print(value+" ");
}
//前序查找
public TreeNode preSearch(int i) {
TreeNode target = null;
//对比当前节点的值
if(this.value == i){
return this;
//当前节点的值不是要查找的节点
}else{
//查找左儿子
if(leftNode!=null){
//有可能可以查到,也可能查不到,查不到的话,target还是一个null
target = leftNode.preSearch(i);
}
//如果不为空,说明在左儿子中已经找到
if(target!=null){
return target;
}
//查找右儿子
if(rightNode!=null){
target = rightNode.preSearch(i);
}
}
return target;
}
//中序查找
public TreeNode midSearch(int i) {
TreeNode target = null;
//查找左儿子
if(leftNode!=null){
//有可能可以查到,也可能查不到,查不到的话,target还是一个null
target = leftNode.midSearch(i);
}
//如果不为空,说明在左儿子中已经找到
if(target!=null){
return target;
}
//对比当前节点的值
if(this.value==i){
return this;
}
//查找右儿子
if(rightNode!=null){
target = rightNode.midSearch(i);
}
return target;
}
//后序查找
public TreeNode afterSearch(int i) {
TreeNode target = null;
//查找左儿子
if(leftNode!=null){
//有可能查到,有可能查不到,查不到的话,target还是一个null
target = leftNode.afterSearch(i);
}
//如果不为null,说明在左儿子中已经找到
if(target!=null){
return target;
}
//查找右儿子
if(rightNode!=null){
target = rightNode.afterSearch(i);
}
//如果target不为空,说明在右儿子中已经找到
if(target!=null){
return target;
}
//对比当前节点的值
if(this.value==i){
return this;
}
return null;
}
//删除一个子树
public void delete(int i) {
TreeNode parent = this;
//判断左儿子
if(parent.leftNode!=null&&parent.leftNode.value==i){
parent.leftNode=null;
return;
}
//判断右儿子
if(parent.rightNode!=null&&parent.rightNode.value==i){
parent.rightNode=null;
return;
}
//递归检查并删除左儿子
parent = leftNode;
if(parent!=null){
parent.delete(i);
}
//递归检查并删除右儿子
parent = rightNode;
if(rightNode!=null){
rightNode.delete(i);
}
}
}
package demo5;
public class TestBinaryTree {
public static void main(String[] args) {
//创建一棵树
BinaryTree binaryTree = new BinaryTree();
//创建一个根节点
TreeNode root = new TreeNode(1);
//把根节点赋给树
binaryTree.setRoot(root);
//创建一个左节点
TreeNode rootL = new TreeNode(2);
//把新创建的节点设置为根节点的子节点
root.setLeftNode(rootL);
//创建一个右节点
TreeNode rootR = new TreeNode(3);
//把新创建的节点设置为根节点的子节点
root.setRightNode(rootR);
//为第二层的左节点创建两个子节点
rootL.setLeftNode(new TreeNode(4));
rootL.setRightNode(new TreeNode(5));
//为第二层的右节点创建两个子节点
rootR.setLeftNode(new TreeNode(6));
rootR.setRightNode(new TreeNode(7));
//前序遍历树
System.out.print("前序遍历:");
binaryTree.frontShow();
System.out.println();
//中序遍历树
System.out.print("中序遍历:");
binaryTree.midShow();
System.out.println();
//后序遍历树
System.out.print("后序遍历:");
binaryTree.afterShow();
System.out.println();
//前序查找
System.out.println("========================");
System.out.println("前序查找:");
TreeNode res1 = binaryTree.preSearch(3);
System.out.println(res1);
//中序查找
System.out.println("中序查找:");
TreeNode res2 = binaryTree.midSearch(5);
System.out.println(res2);
//后序查找
System.out.println("后序查找:");
TreeNode res3 = binaryTree.afterSearch(3);
System.out.println(res3);
//删除一棵子树
System.out.println("删除节点前:");
binaryTree.frontShow();
System.out.println();
System.out.println("删除节点后:");
binaryTree.delete(4);
binaryTree.frontShow();
}
}