二叉树的后继节点

一 题目

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

二 例子

对于下面的二叉树，8的后继节点为9(题目说的中序遍历)，6的后继节点为7，5的后继节点为6

三 思路

思路很简单，

①如果节点有右子树，则该节点的后继节点就是往右子树出发，然后转到右子树的左子树，一直到左子树的左子树为空（即输入节点的右子树的最左子树，例如节点8，后继节点就是9）

②如果节点没有右子树，则向上寻找父节点，直到父节点的左子树等于当前节点，则该父节点就是后继节点（如节点7，没有右子树，则向上找，这时到6这个节点，因为6的左子树不等于7，所以继续向上找，这时到节点8，8的左子树等于当前节点6，所以返回8）

四 程序源码

1. class TreeLinkNode {
2. int val;
3. TreeLinkNode left = null;
4. TreeLinkNode right = null;
5. TreeLinkNode next = null;
7. TreeLinkNode(int val) {
8. this.val = val;
9. }
10. }
11. */
12. public class Solution {
13. public TreeLinkNode GetNext(TreeLinkNode pNode){
14. if(pNode==null) return null;
15. if (pNode.right != null) {
16. pNode = pNode.right;
17. while(pNode!=null&&pNode.left!=null){
18. pNode = pNode.left;
19. }
20. return pNode;
21. }
22. while(pNode.next!=null){
23. if(pNode.next.left == pNode) return pNode.next;
24. pNode = pNode.next;
25. }
26. return null;
27. }
28. }