- 题干:给定一颗二叉树和其中一个节点,找出中序遍历的下一个节点。
- 思路:根据中序遍历去查找,设置一个flag去标记已经遍历的结点是否是给定的结点。我在这里借用了自己之前写的类(参见添加链接描述),所以没有指向父结点的指针。代码如下:
public BinaryTreeNode findNext(BinaryTree bTree , BinaryTreeNode target) {
BinaryTreeNode result = null;
BinaryTreeNode node = bTree.getRoot();
boolean flag = false;
Stack<BinaryTreeNode> stack = new Stack<>();
while(node!=null||stack.size()!=0) {
while(node!=null) {
stack.push(node);
node = node.getLeft();
}
if(stack.size()!=0) {
if(flag==false) {
if(stack.peek().equals(target)) {
flag = true;
}
node = stack.pop().getRight();
} else {
//找到要找的结点
result = stack.peek();
break;
}
}
}
return result;
}
- 如果是有指向父节点的指针,根据中序遍历的过程可知:
- 如果有右孩子,则获取右子树中最左边的节点;
- 如果没有右孩子且为父节点的左节点,则为该点的父节点;
- 如果没有右孩子且为父节点的右结点,则沿着该节点的父节点向上查找,直到找到某一点是其父节点的左孩子,这个父节点即为答案。也就是说,要找到使其成为左子树中一员的祖父辈节点。