二叉树的下一个节点

来源:互联网 发布:mac的flash屏蔽 编辑:程序博客网 时间:2024/05/17 19:18

题目描述:

给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。

思路分析:

首先判断该节点的右子树是否为空,将该节点的右子树记为R

1 若R非空

判断R的左子树是否为空,若R的左子树为空,则该节点的中序遍历的下一个节点为R,若R的左子树不为空,则该节点的中序遍历的下一个节点为R的左子树的最左节点,

2 若R为空

则中序遍历该节点的下一个节点是其最近的一个根节点,且满足该节点在根节点的左子树上。

代码如下

/*
public class TreeLinkNode {
    int val;
    TreeLinkNode left = null;
    TreeLinkNode right = null;
    TreeLinkNode next = null;


    TreeLinkNode(int val) {
        this.val = val;
    }
}
*/
public class Solution {
    public TreeLinkNode GetNext(TreeLinkNode pNode)
    {                                                                                                                                                                       
        if(pNode.right!=null){
            TreeLinkNode l=pNode.right.left;
            if(l==null)
                return pNode.right;
            else{
                while(l.left!=null){
                    l=l.left;
                }
                return l;
            }
        }
        else{
            while(pNode.next!=null&&pNode.next.left!=pNode){
                pNode=pNode.next;
            }
            if(pNode.next==null)
                return null;
            else
            return pNode.next;
        }
    }
}


原创粉丝点击