《剑指offer》-- 二叉树的下一个结点

题目描述：

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

解题思路：

/**

* 思路：

*

* 对于一个节点，其后续节点有三种形式：

* 1. 该节点的右指针为null，那么就往前找其父节点（设为a），

* 若a是其父节点的左子节点，那么下一个节点就是a的父节点，否则，就一直往上找，找到该节点为其父节点的左节点位置；

* 如果一直找到了跟节点（即其父节点为null），那么久返回null

* 2. 该节点是的右指针不为null，那么找其右子树的第一个节点

*

*/

本道题目不难，树遍历基本的三种方式，本道题的以上思路，可以综合在一个代码当中完成，也可以分功能到不同的函数内部，这是两种小细节的处理方式，我是使用的第二种，即在寻找右子树中序遍历下的第一个节点时，递归调用。

code:

/**
* 题目描述
*
* 给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。
*
* date: 2016.4.2   10:53
* @author SSS
*
*/
public class Solution {

public TreeLinkNode GetNext(TreeLinkNode pNode) {
TreeLinkNode resultNode  = null;

// 总体情况分为该节点右指针是否为0，若不为null，那么其下一个节点一定在右子树上，否则就要往上寻找
if (pNode.right != null) {
resultNode = getFirstNodeOfInOrder(pNode.right);

} else {    // 往上找
TreeLinkNode parentNode = pNode.next;
TreeLinkNode tempNode = pNode;
while(parentNode != null) {
if (parentNode.left == tempNode) {  // 该节点为父节点的左节点，那么父节点就是中序遍历的下一个节点
resultNode = parentNode;
break;
} else { // 否则就一直往上遍历
tempNode = parentNode;
parentNode = parentNode.next;
}
}
// 这种情况是一直遍历到整个树的根节点
if (parentNode == null) {
resultNode = null;
}

}

return resultNode;
}

/**
* 在中序遍历下，找到该节点的下一个节点,前提是当前节点是非空
* @param node
* @return
*/
public TreeLinkNode getFirstNodeOfInOrder(TreeLinkNode node) {
TreeLinkNode resultNode  = node;
if(node.left == null) {
return node;
}

resultNode = getFirstNodeOfInOrder(node.left);

return resultNode;
}
/*
public static void main(String []args) {

Solution solution = new Solution();

TreeLinkNode rootNode = new TreeLinkNode(4);
TreeLinkNode aNode = new TreeLinkNode(23);
TreeLinkNode bNode = new TreeLinkNode(12);
rootNode.left = aNode;
rootNode.right = bNode;
aNode.next = rootNode;
bNode.next = rootNode;

TreeLinkNode cNode = new TreeLinkNode(1);
TreeLinkNode dNode = new TreeLinkNode(65);
aNode.left = cNode;
aNode.right = dNode;
cNode.next = aNode;
dNode.next = aNode;

TreeLinkNode eNode = new TreeLinkNode(234);
TreeLinkNode fNode = new TreeLinkNode(64);
bNode.left = eNode;
bNode.right = fNode;
eNode.next = bNode;
fNode.next = bNode;

TreeLinkNode resultNode = solution.GetNext(bNode);

if (resultNode != null) {
System.out.println(resultNode.val);
} else {
System.out.println("null");
}

}*/
}