欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Java查找二叉树中序遍历节点的下一个节点,二叉树节点使用泛型类

程序员文章站 2024-03-14 21:56:17
...

接上一篇博客,直接上代码吧

节点类

package swordoffer.binarytree;

/**
 * @program: javaStudyTest
 * @description: 二叉树与二叉树重建
 * @create: 2020-03-03
 **/
public class BinaryTreeNode<E> {
    //数据部分
    private E item;
    //父亲节点
    BinaryTreeNode<E> fatherNode;
    //左子节点
    BinaryTreeNode<E> leftChildNode;
    //右子节点
    BinaryTreeNode<E> rightChildNode;


    public BinaryTreeNode(E item) {
        this.item = item;
    }
}

二叉树工具类,查找下一个节点算法

package swordoffer.binarytree;

import org.apache.commons.lang3.ArrayUtils;
import java.util.Arrays;

/**
 * @program: javaStudyTest
 * @description: 重建二叉树
 * @create: 2020-03-03
 **/
public class BinaryTreeUtils<T> {
	/**
     * 查找二叉树中的中序遍历的下一个节点(有子节点指向父节点指针的情况)
     *
     * @param node
     * @return
     */
    private BinaryTreeNode<T> getNextNode(BinaryTreeNode<T> node) {
        if (node == null) return null;
        BinaryTreeNode<T> nextNode = null;

        //节点存在右子树
        if (node.rightChildNode != null) {
            BinaryTreeNode<T> tempNode = node.rightChildNode;
            while (tempNode.leftChildNode != null) {
                tempNode = tempNode.leftChildNode;
            }
            nextNode = tempNode;
        } else if (node.fatherNode != null) { //不存在右子树,存在父节点
            //如果该节点是其父节点的左子节点
            if (node.fatherNode.leftChildNode == node) {
                nextNode = node.fatherNode;
            } else { //该节点是其父节点的右子节点
                BinaryTreeNode<T> tempNode = node.fatherNode;
                while (tempNode.fatherNode != null && tempNode.fatherNode.rightChildNode == tempNode) { //该节点存在父节点,且为该父节点的右子节点
                    tempNode = tempNode.fatherNode;
                }
                nextNode = tempNode.fatherNode; // 因为父节点为空而退出循环,或因为是父节点的左子节点而退出循环
            }
        }
        return nextNode;
    }
}