给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
解法一:根据提供条件节点中包含父节点我们可以先找出根节点,然后中序遍历找出下一个节点。
import java.util.*;
public class Solution {
ArrayList<TreeLinkNode> list = new ArrayList<TreeLinkNode>();
public TreeLinkNode GetNext(TreeLinkNode pNode)
{
TreeLinkNode rootNode = pNode;
while(rootNode.next!=null){
rootNode = rootNode.next;
}
getNode(rootNode);
for(int i = 0;i<list.size();i++){
if(pNode==list.get(i)){
return (list.size()-1)==i?null:list.get(i+1);
}
}
return null;
}
public void getNode(TreeLinkNode node){
if(node!=null){
getNode(node.left);
list.add(node);
getNode(node.right);
}
}
}
时间复杂度O(n);
空间复杂度:O(n);
解法二:直接找下一节点,拆分情况,已知条件下,存在三种条件(这里盗一张图中序遍历为:{D,B,H,E,I,A,F,C,G})。
情况一:其下存在右子树,我们只需找到其下右子树的节点下的最左节点即是我们所需要的下一节点如B我们所对应的的目标节点为H。
情况二:其下不存在右子树,但是该节点为父节点的左子树节点,如D我们对应的目标节点为B即其父节点。
情况三:其下不存在右子树,该节点也不是父节点的左子树节点,我们寻找其父节点,若父节点的下一个节点的左节点为父节点则父节点的下一个节点为我们的目标节点,若不是,则按照该规则继续向该节点的父节点寻找。如I节点所对应的的目标节点为A,G没有我们定义的节点,所以没有我们的目标节点。
public
class
Solution {
public
TreeLinkNode GetNext(TreeLinkNode pNode) {
// 1.
if
(pNode.right !=
null
) {
TreeLinkNode pRight = pNode.right;
while
(pRight.left !=
null
) {
pRight = pRight.left;
}
return
pRight;
}
// 2.
if
(pNode.next !=
null
&& pNode.next.left == pNode) {
return
pNode.next;
}
// 3.
if
(pNode.next !=
null
) {
TreeLinkNode pNext = pNode.next;
while
(pNext.next !=
null
&& pNext.next.right == pNext) {
pNext = pNext.next;
}
return
pNext.next;
}
return
null
;
}
}
时间复杂度:O(n);
空间复杂度:O(1);
练习地址:https://www.nowcoder.com/practice/9023a0c988684a53960365b889ceaf5e