今日你如约而至了吗?每天坚持至少打卡一道,保持正常的手感。今天还是一道二叉树的题,哈哈,没办法,全剩下链表或者二叉树的题了,不做也得做,兄嘚,硬着头皮上啊。上题:
题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
方法一:常规法
分析:需要按照中序遍历的顺利找到给定节点的下一个节点,那我们需要得到中序遍历的结果,但是各种遍历都是从头开始的,所以在遍历之前需要找到头结点(题目中告诉了你它有父节点指针,这个就是为了找头结点的。)最后遍历,找到当前结点的下一个结点,完事。
步骤:
1、根据给定结点,利用父指针找到根结点;
2、根据根结点,递归中序遍历结果;
3、遍历中序结果,得到当前结点的下一个结点。
代码:
/*
public class TreeLinkNode {
int val;
TreeLinkNode left = null;
TreeLinkNode right = null;
TreeLinkNode next = null;
TreeLinkNode(int val) {
this.val = val;
}
}
*/
import java.util.ArrayList;
public class Solution {
ArrayList<TreeLinkNode> arr = new ArrayList<>();
public TreeLinkNode GetNext(TreeLinkNode pNode)
{
if(pNode==null)return null;//这里不能判断if(pNode.next==null),因为它可能是叶子节点,但是中序遍历后,它是有可能有后面结点的。
TreeLinkNode root=null;
TreeLinkNode temp=pNode;//不要懂给定的结点
while(temp!=null){ //根据while循环,找到根结点
root=temp;;
temp=temp.next;
}
mid(root); //中序遍历
int len=arr.size();
for(int i=0;i<len;i++){ //找到给定结点的下一个结点。
if((arr.get(i)==pNode)&&(i+1!=len)){
return arr.get(i+1);
}
}
return null;
}//中序遍历的递归实现。
public void mid(TreeLinkNode root){
if(root==null)return;
mid(root.left);
arr.add(root);
mid(root.right);
}
}
牛客运行通过
运行时间:27ms
运行内存:9836Kb
这是按照套路走的,有按照套路的方法,那么就有不按照套路的方法。
方法二:不按套路法
这是牛客官方给的答案,各位可以参考,因为需要分析。链接中讲的非常详细,它是用C++实现的,我用java重新操作了一遍。
https://blog.nowcoder.net/n/737a842c39d6473db2a10d4f1de7610c?f=comment
public class Solution {
public TreeLinkNode GetNext(TreeLinkNode pNode){
if(pNode==null)return pNode;
if(pNode.right!=null){
pNode = pNode.right;
while(pNode.left!=null){
pNode=pNode.left;
}
return pNode;
}
while(pNode.next!=null){
TreeLinkNode root=pNode.next;
if(root.left==pNode){
return root;
}
pNode=pNode.next;
}
return null;
}
}
牛客运行通过
运行时间:24ms
运行内存:9672Kb
方法一一般人都可以想到,但方法二比较难想,但是遇到链表或者二叉树的题,画个示意图,理解会更快。
欢迎各位互相学习和交流哦~~