二叉树的下一个结点

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

分析

  • 如果一个节点有右子树,那么中序遍历的下一个节点就是它的右子树中的最左子节点
  • 如果一个节点没有右子树,且该节点是其父节点的左子节点,那么它的下一个节点就是它的父节点
  • 如果一个节点没有右子树,且它还是父节点的右子节点,那么我们可以沿着指向父节点的指针一直向上遍历,直到找到一个是它父节点的左子节点的节点,如果这样的节点存在,那么这个节点的父节点就是我们要找的下一个节点。

节点类:

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 == null) {
    		return null;
    	}
    	
    	TreeLinkNode node = null;
    	
    	if(pNode.right != null) {
    		TreeLinkNode nodeRight = pNode.right;
    		while(nodeRight.left != null) {
    			nodeRight = nodeRight.left;
    		}
    		
    		node = nodeRight;
    	}
    	else if(pNode.next != null) {
    		TreeLinkNode current = pNode;
    		TreeLinkNode parent = pNode.next;
    		
    		while(parent != null && current == parent.right) {
    			current = parent;
    			parent = parent.next;
    		}
    		
    		node = parent;
    	}
    	
    	return node;
    	
    }
}
posted @ 2019-04-09 22:14  如是说  阅读(629)  评论(0编辑  收藏  举报