题目:
给定一颗二叉树和其中一个节点,如何找出中序遍历序列的下一个节点?树中的节点除了有两个分别指向左、右子节点的指针,还有一个指向父节点的指针。二叉树节点定义如下:
class BinaryTreeNode {
int data;
BinaryTreeNode left;
BinaryTreeNode right;
BinaryTreeNode parent;
BinaryTreeNode(int data) {
this.data = data;
}
}
算法思想:
1.如果该节点有右子树,那么它的下一个节点就是它的右子树的左子节点。
2.如果该节点没有右子树,并且是它的父节点的左子节点,那么它的下一个节点就是它的父节点。
3..如果该节点没有右子树,并且是它的父节点的右子节点,那么它的下一个节点就是它的第一个有左子节点的祖先节点(不包含父节点);否则没有下一个节点。
代码实现:
public BinaryTreeNode getNext(BinaryTreeNode pNode) {
if(pNode == null)
return null;
if(pNode.right != null) { //节点有右子树
BinaryTreeNode node = pNode.right; //得到右子节点
while(node.left != null)
node = node.left;
return node;
} else { //节点没有右子树
while (pNode.parent != null) {
BinaryTreeNode node = pNode.parent; //得到父节点
if (node.left == pNode) //该节点是父节点的左子节点
return node;
pNode = pNode.parent; //得到父节点的父节点
}
}
return null;
}