《剑指offer》8.二叉树的下一个节点

题目描述

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

思路：如果有右子树，则找右子树的最左节点；如果没有右子树，则向上递归找该节点属于哪个父节点左支，则返回该父节点

/*
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;
// 如果有右子树，则找右子树的最左节点
if (pNode.right != null) {
pNode = pNode.right;
while (pNode.left != null)
pNode = pNode.left;
return pNode;
}

// 如果没有右子树，则找属于某父节点左支的该父节点
while (pNode.next != null) {
if (pNode == pNode.next.left)
return pNode.next;
else pNode = pNode.next;
}
return null;
}
}

运行时间：32ms

占用内存：9200k