Java-二叉树-插入、删除、遍历

class Node {
    private int data;
    private Node leftChild;
    private Node rightChild;
    private Node parent;

    public Node getParent() {
        return parent;
    }

    public void setParent(Node parent) {
        this.parent = parent;
    }

    public Node(int data, Node leftChild, Node rightChild, Node parent) {
        this.data = data;
        this.leftChild = leftChild;
        this.rightChild = rightChild;
        this.parent = parent;

    }

    public int getData() {
        return data;
    }

    public void setData(int data) {
        this.data = data;
    }

    public Node getLeftChild() {
        return leftChild;
    }

    public void setLeftChild(Node leftChild) {
        this.leftChild = leftChild;
    }

    public Node getRightChild() {
        return rightChild;
    }

    public void setRightChild(Node rightChild) {
        this.rightChild = rightChild;
    }

}

public class BinaryTree {
    private Node root;//根节点
    //插入节点
    public void insertNode(Node root, Node node) {

        Node current = root;
        while (true) {
            if (node.getData() < current.getData()) {
                if (current.getLeftChild() == null) {
                    node.setParent(current);
                    current.setLeftChild(node);
                    break;
                } else {
                    current = current.getLeftChild();
                }
            } else {
                if (current.getRightChild() == null) {
                    node.setParent(current);
                    current.setRightChild(node);
                    break;
                } else {
                    current = current.getRightChild();
                }

            }
        }
    }
    //删除节点
    public void deleteNode(Node node) {
        if (node.equals(root)) {
            root = null;
        } else if (node.getParent() != null) {
            if (node == node.getParent().getLeftChild()) {
                node.getParent().setLeftChild(null);
            } else {
                node.getParent().setRightChild(null);

            }
        }
    }

    //获取某节点的高度
    public int geHeight(Node node) {
        if (node == null) {
            return 0;
        } else {
            int leftHeight = geHeight(node.getLeftChild());
            int rightHeight = geHeight(node.getRightChild());
            int max = Math.max(leftHeight, rightHeight);
            return max + 1;
        }
    }

    //获取某节点的子节点个数
    public int getChildNodes(Node node) {
        if (node == null) {
            return 0;
        } else {
            int leftNodes = getChildNodes(node.getLeftChild());
            int rightNodes = getChildNodes(node.getRightChild());
            return leftNodes + rightNodes + 1;
        }
    }

    //先序遍历树
    public void PreOrder(Node root) {
        if (root == null)
            return;
        System.out.print(root.getData() + " ");
        PreOrder(root.getLeftChild());
        PreOrder(root.getRightChild());
    }

    //中序
    public void MidOrder(Node root) {
        if (root == null) return;
        MidOrder(root.getLeftChild());
        System.out.print(root.getData() + " ");
        MidOrder(root.getRightChild());
    }

    //后序
    public void LastOrder(Node root) {
        if (root == null) return;
        LastOrder(root.getLeftChild());
        LastOrder(root.getRightChild());
        System.out.print(root.getData() + " ");

    }

    public BinaryTree() {

    }

    public BinaryTree(Node root) {
        this.root = root;
    }

    public Node getRoot() {
        return root;
    }

    public void setRoot(Node root) {
        this.root = root;
    }

}

public class Main {
    public static void main(String[] args) {
        BinaryTree bt = new BinaryTree(new Node(1, null, null, null));
        int a[] = {5, 3, 2, 7, 4, 9, 8};
        for (int i = 0; i < 7; i++) {
            bt.insertNode(bt.getRoot(), new Node(a[i], null, null, null));
        }
//        System.out.println(bt.geHeight(root));//高度
//        bt.PreOrder(root);
//        System.out.println();
//        bt.MidOrder(root);
//        System.out.println();
//        bt.LastOrder(root);
//        System.out.println();
//        bt.deleteNode(bt.getRoot());
//        bt.PreOrder(bt.getRoot());
 //       System.out.println(bt.getChildNodes(bt.getRoot()));//子节点数

    }
}