给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针
解题思路:
(1)node有右子树,下一个节点是有子树的最左的一个节点
(2)node没有右子树,是父节点的左子树,下一个节点是父节点
(3)node没有右子树,是父节点的右子树,需要向上一直寻找到该子树的是某个节点的左子树,下一个节点是该节点
java
/*
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 root) {
if (root == null) {
return null;
}
TreeLinkNode node = null;
if (root.right != null) {
node = root.right;
while (node.left != null) {
node = node.left;
}
} else if (root.next != null) {
TreeLinkNode curr = root;
TreeLinkNode parent = root.next;
while (parent != null && curr == parent.right) {
curr = parent;
parent = parent.next;
}
node = parent;
}
return node;
}
}