给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
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 pNode)
{
if(pNode == null){
return null;
}
if(pNode.right != null){
return getNextNode(pNode.right);
}
if(pNode.next == null){
return null;
}
if(pNode.next.left == pNode){
return pNode.next;
}
while(pNode.next.right == pNode){
pNode = pNode.next;
if(pNode.next == null){
return null;
}
}
return pNode.next;
}
public TreeLinkNode getNextNode(TreeLinkNode pNode){
if(pNode == null){
return null;
}
TreeLinkNode result = null;
result = getNextNode(pNode.left);
if(result == null)
result = pNode;
return result;
}
}
- 中序遍历是直先遍历节点的左子树,然后是节点本身,然后是节点的右子树。
- 前序遍历是先是节点本身,然后左子树,再之后是右子树。
- 后序遍历是先左子树,后右子树,最后是节点本身
- 在子树中的遍历规则同上
他说要找到中序遍历下的下一个节点。这个节点可以分为两种情况
- 该节点有右子树
- 该节点没有右子树
- 第一种比较处理起来比较简单,直接将其右节点进行中序遍历即可,并将一个遍历到的最右节点返回。
- 第二种情况又分为两种情况
- 该节点是父节点的左子节点
- 该节点是父节点的右子节点
- 这里,该节点是父节点的左子节点的这种情况比较简单,直接将父节点返回即可
- 如果是父节点的右子节点的话,需要不断的向上移动,直到对应的节点不是父节点的右节点(即左节点),,既然他是父节点的左节点,此时将这个节点父节点返回即可,或者遍历到了根节点,返回null;
所以需要引入一个进行中须遍历并反会第一个遍历到的节点的函数即上面的getNextNode
因为这个代码主要是根据思路进行分解得来的,先用递归简单的实现了一下。
如果不去嵌套其他函数,使用while来进行求解,代码会简洁许多
如下
public class Solution {
public TreeLinkNode GetNext(TreeLinkNode pNode)
{
if(pNode == null){
return null;
}
if(pNode.right != null){
pNode = pNode.right;
while(pNode.left != null){
pNode = pNode.left;
}
return pNode;
}
while(pNode.next != null){
if(pNode.next.left== pNode){
return pNode.next;
}
pNode = pNode.next;
}
return null;
}
}