57.二叉树的下一个结点
题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
题目解答
/* 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 pNode; } //1.pNode有右子树->pNode.right的最左节点 if(pNode.right!=null){ pNode=pNode.right; while(pNode.left!=null) pNode=pNode.left; return pNode; } while(pNode.next!=null){ //2.pNode没有右子树,pNode是父节点的左子节点->父亲节点 if(pNode.next.left==pNode) return pNode.next; //3.pNode没有右子树,pNode是父节点的右子节点->父亲节点的下一个节点 pNode=pNode.next; } //4.没有下一个节点 return null; } }