牛客网:二叉树的下一个节点
题目描述:
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时
包含指向父结点的指针。
解题思路
有如下情况:
1.如果一个节点的右子树不为空,那么该节点的下一个节点是右子树的最左节点;
(中序遍历的过程:先遍历树的左子树,再遍历根节点,最后再遍历右子树。所以最左节点是中序遍历的第一个节点。)
2.如果一个节点的右子树不为空,向上找第一个左链接指向的树包含该节点的(祖先节点)。
(如果一个节点没有右子树,且节点是她父节点的左子树,那么他的下一个节点就是他的父节点)
(如果一个节点既没有右子树,且他还是他父节点的右子节点,可以沿着指向父节点的指针一直
向上遍历,直到找到一个是他父节点的左子节点的节点,如果这样的节点存在,那么这个节点的
父节点就是我们要找的下一个节点)
// 二叉树中序遍历
void traverse(TreeLinkNode root) {
if (root == null) return;
traverse(root.left);
visit(root);
traverse(root.right);
}
代码:
public TreeLinkNode GetNode(TreeLinkNode pNode) {
if (pNode.right != null) {
TreeLinkNode node = pNode.right;
while (node.left != null) {
node = node.left;
}
return node;
} else {
while (pNode.next != null) {
TreeLinkNode parent = pNode.next;
if (parent.left == pNode) {
return parent;
}
pNode = pNode.next;
}
}
return null;
}