面试题58. 二叉树的下一个节点

面试题58. 二叉树的下一个节点

题目描述

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

比如下图中，中序遍历是【d、b、h、e、i、a、f、c、g】



根据中序遍历，我们能发现这样的规律：

如果一个结点有右子树，那么它的下一个结点就是右子树的最左结点

比如 a有右子树，a的下一个结点是右子树的最左，也就是f

比如 b有右子树，b的下一个结点是h

如果一个结点没有右子树，这种情况需要通过循环，从当前结点开始沿着父指针向上找，直到找到一个是左孩子的结点，这个左孩子结点的父节点，就是下一个结点。

比如 h，它是e的左孩子，所以h的下一个结点就是e

比如 i，它是e的右孩子，所以从i开始向上找直到找到一个左孩子。由于e是b的右孩子，所以需要继续向上找。直到找到了b（b是a的左孩子），所以返回a。

根据上面的算法，就能写出代码：

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;

// 如果pNode有右子树，则下一个节点是右子树的最左
if(pNode.right != null) {
TreeLinkNode node = pNode.right;
while(node.left != null) { // 循环找最左
node = node.left;
}
return node;
}

// 如果pNode没有右子树，则
// 把pNode作为当前节点，向上找父节点，如果当前节点是父节点的左孩子，则返回父节点
// 否则把pNode的父节点作为当前节点，向上找当前节点的父节点，直到当前节点是父节点的左孩子
TreeLinkNode node = pNode;
while(node.next != null && node.next.left != node) {
node = node.next;
}
return node.next;
}
}