数据结构(Java):力扣 二叉树面试OJ题(二)【进阶】
目录
💎 1、题一:二叉树的层序遍历
层序遍历, 即从上到下,从左到右,一层一层的将数据遍历一遍。
🌟 1.1 思路1(递归求解)
- 首先,根据上篇求树高度的方法,求出二叉树的高度。接着,实例化出高度个一维顺序表添加到二维顺序表中。
- 记录好每层数据的层数,根据层数并按顺序来将来将数据添加到二维顺序表的第几行中(记录层数来放入数据,是解题关键点!!!)。
- 默认根节点是第0层,与顺序表下标相对应。
- 这里以前序遍历的思想递归,添加各层数据。
🌟 1.1.1 思路1代码
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> listVal = new ArrayList<>();
if(root == null) {
//注意:若根节点为空,要返回空的顺序表
return listVal;
};
int height = treeHeight(root);
for(int i = 0; i < height; i++) {
List<Integer> list = new ArrayList<>();
listVal.add(list);
}
//根节点的层数默认为0
addNode(root,0,listVal);
return listVal;
}
//求树高度
public int treeHeight(TreeNode root) {
if(root == null) return 0;
return Math.max(treeHeight(root.left),treeHeight(root.right)) + 1;
}
public void addNode(TreeNode root,int level,List<List<Integer>> listVal) {
if(root == null) {
return;
}
//以前序遍历的思想,根据层数来将数据添加到该行的顺序表中
listVal.get(level).add(root.val);
addNode(root.left,level+1,listVal);
addNode(root.right,level+1,listVal);
}
}
🔆 1.2 思路2(队列求解)
- 首先创建一个队列,将根节点放入队列当中
- 每次循环前记录队列中元素个数size,依次出队size个元素,每次出队的元素(节点)都要判断其左右子树是否为空,若不为空则添加到队列中
- 将每次出队的元素的val值添加到一维顺序表中,size个元素全部出队完后,将这个一维顺序表添加到二维顺序表中,也就是说,每size个出队的元素都是同一层的节点
- 循环以上过程,直至元素全部添加至二维顺序表中(队列为空)
🔆 1.2.1 思路2代码
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> listVal = new ArrayList<>();
if (root == null) {
//注意:若根节点为空,要返回空的顺序表
return listVal;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
//队列空时,说明节点已全部添加到二维顺序表中
while (!queue.isEmpty()) {
int size = queue.size();
//存储每一层节点的顺序表
List<Integer> list = new ArrayList<>();
//将队列中的size个元素依次出队,
//并判断其左右子树是否为空,若不为空则添加到队列中
while (size != 0) {
TreeNode out = queue.poll();
list.add(out.val);
if (out.left != null) {
queue.offer(out.left);
}
if (out.right != null) {
queue.offer(out.right);
}
size--;
}
//将每一层的一维顺序表添加到二维顺序表中
listVal.add(list);
}
return listVal;
}
}
💎 2、题二:判断是否为完全二叉树
本题无链接,
题目即:给出一棵树的根节点,判断是否为完全二叉树。
🌟 2.1 思路分析
这题的解题思路和上一题队列求解差不多,都是借助队列这一数据结构:
- 首先,创建队列,放入根节点。
- 将一个元素出队,cur接收出队元素,判断cur是否为空,若cur不为空,则将其左右孩子放入队列中(若左右子树为null,也要将null入队),循环这一过程。
- 当cur为null时结束循环(出队的是null,cur接收,此时cur为null),此时,只要判断队列中的剩余元素是否存在非null元素:若剩余元素全为null则说明该树为完全二叉树;若剩余元素存在非null元素则说明该树不为完全二叉树。
🌟 2.2 代码
boolean isCompleteTree(TreeNode root) {
if(root == null) {
return true;
}
//创建队列 添加根节点
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
//当cur不为空时,添加其左右孩子
TreeNode cur = root;
while (cur != null) {
cur = queue.poll();
if (cur != null) {
queue.offer(cur.left);
queue.offer(cur.right);
}
}
//cur为空时,判断队列中是否有非空元素
//若剩余元素全为null则说明该树为完全二叉树;
// 若剩余元素存在非null元素则说明该树不为完全二叉树。
boolean flg = false;
while (!queue.isEmpty()) {
TreeNode cur2 = queue.poll();
if (cur2 != null) {
flg = true;
}
}
if (flg == true) {
return false;
}else {
return true;
}
}
💎 3、题三:求二叉树中节点的最近公共祖先
🌟 3.1 思路分析
- ①:当p、q在根节点的左右子树上时,祖先就是这个根节点
- ②:当p、q都在根节点的右子树上时,由于节点的祖先可以是它自己,所以祖先就是p或q本身
- ③:当p、q都在根节点的左子树上时,由于节点的祖先可以是它自己,所以祖先就是p或q本身
- 以前序遍历思想递归,若遇见p、q节点便返回
- 若左右子树都有返回值,说明祖先就是这个根节点;(对于第一层的根节点)若只有左子树有返回值,说明p、q均在左子树上,返回的节点就是最近祖先;(对于第一层的根节点)若只有右子树有返回值,说明p、q均在右子树上,返回的节点就是最近祖先。
🌟 3.2 代码
class Solution {
public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
return findNode( root, p, q);
}
public TreeNode findNode(TreeNode root, TreeNode p, TreeNode q) {
if(root == null) {
return null;
}
//先判断根节点是否为p、q节点
if(root.val == p.val || root.val == q.val) {
return root;
}
//查找左右子树
TreeNode leftNode = findNode(root.left,p,q);
TreeNode rightNode = findNode(root.right,p,q);
//若左右子树都有返回值,说明祖先就是这个根节点
if(leftNode != null && rightNode != null) {
return root;
}
//在左树上找到了
if(leftNode != null && rightNode == null) {
return leftNode;
}
//在右树上找到了
if(rightNode != null && leftNode == null) {
return rightNode;
}
return null;
}
}
💎 4、题四:根据二叉树创建字符串
🌟 4.1 思路分析
- 借助StringBuffer拼接字符串,最后返回字符串
- 以前序遍历思想递归
- 拼接当前根节点,判断根节点的左孩子是否为空,若左孩子不为空则拼接"(",并递归左子树;若左孩子为空且右孩子为空,则直接返回即可;若左孩子为空但右孩子不为空,注意!!!此时应该拼接"()",继续递归右子树。
- 若右孩子为空,则直接返回;若右孩子不为空,则拼接"("并递归右子树。
- 在根节点的左右子树完成即递归回退后拼接相应的")"
- 调用toString方法返回字符串
🌟 4.2 代码
class Solution {
StringBuilder stringBuilder = new StringBuilder();
public String tree2str(TreeNode root) {
tree2strChild(root);
return stringBuilder.toString();
}
public void tree2strChild(TreeNode root) {
if(root == null) {
return ;
}
stringBuilder.append(root.val);
if(root.left != null) {
//左子树不为空,则必然添加"("
stringBuilder.append("(");
//递归左子树
tree2strChild(root.left);
//该节点左子树完,则拼接")"
stringBuilder.append(")");
}else {
//左子树为空,且右子树为空,直接返回
if(root.right == null) {
return;
}else {
//左子树为空,但右子树不为空,需拼接"()"
stringBuilder.append("()");
}
}
//若右孩子为空,则直接返回;若右孩子不为空,则拼接"("并递归右子树
if (root.right != null) {
stringBuilder.append("(");
tree2str(root.right);
stringBuilder.append(")");
}else {
return;
}
}
}
💎 5、题五:二叉树前序遍历递归实现与非递归实现
🌟 5.1 递归实现->思路分析
递归实现前序遍历的思想很简单:就是先将根节点的数据放入顺序表中,再将左子树的数据放入顺序表中, 再将右子树的数据放入顺序表中。
🌟 5.1.1 递归实现->代码
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> list = new ArrayList<>();
if(root == null) {
return list;
}
list.add(root.val);
List<Integer> leftList = preorderTraversal(root.left);
list.addAll(leftList);
List<Integer> rightList = preorderTraversal(root.right);
list.addAll(rightList);
return list;
}
}
🔆 5.2 非递归实现->思路分析
非递归实现遍历,我们需要借助一种数据结构--->栈
注意:本题要求将数值放入顺序表中,返回顺序表。故这里访问根节点时,对根节点的操作是将根节点的val值添加到顺序表中。
- 创建栈和顺序表,将根节点赋值给cur
- 将cur入栈并将cur的val值添加到顺序表中,cur更新为cur的left,接着将cur入栈并将cur的val值添加到顺序表中。循环操作,直到cur为空
- 此时cur为空,pop弹出栈顶元素赋给top,将cur更新为top.right;若top.right为空,即更新的cur为空,则继续弹出栈顶元素(top接收),直至cur不为空。
- 此时cur不为空,循环2、3步骤,直至栈中元素为空,说明树中元素已按照前序遍历的方式全部添加到了顺序表中。
🔆 5.2.1 非递归实现->代码
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> list = new ArrayList<>();
Stack<TreeNode> stack = new Stack<>();
TreeNode cur = root;
//注意这里的循环条件
//因为后续cur为空时要弹出栈顶元素,故栈不能为空
//且一开始时栈为空,故cur不为空就可进入循环
//当栈空且cur为空,说明前序遍历已完成
while (cur != null || !stack.isEmpty()) {
//cur不为空就要进栈,并添加进顺序表
//cur = cur.left
while (cur != null) {
stack.push(cur);
list.add(cur.val);
cur = cur.left;
}
//当cur为空时,弹出栈顶元素(top),赋给cur其右子树
//若cur还是为空(top.right为空),循环过来继续弹出元素,并赋给cur新的top.right
TreeNode top = stack.pop();
cur = top.right;
}
return list;
}
}
💎 6、题六:二叉树中序遍历递归实现与非递归实现
中序遍历的实现,不管是在递归实现上还是在非递归实现上,和前序遍历的思想都是相同的,只是访问根节点的时机有所不同。
🌟 6.1 递归实现->代码
class Solution {
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> list = new ArrayList<>();
if(root == null) {
return list;
}
List<Integer> listLeft = inorderTraversal(root.left);
list.addAll(listLeft);
list.add(root.val);
List<Integer> listRight = inorderTraversal(root.right);
list.addAll(listRight);
return list;
}
}
🔆 6.2 非递归实现->代码
class Solution {
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> list = new ArrayList<>();
Stack<TreeNode> stack = new Stack<>();
TreeNode cur = root;
//注意这里的循环条件
//因为后续cur为空时要弹出栈顶元素,故栈不能为空
//且一开始时栈为空,故cur不为空就可进入循环
//当栈空且cur为空,说明前序遍历已完成
while (cur != null || !stack.isEmpty()) {
//cur走左子树
while (cur != null) {
stack.push(cur);
cur = cur.left;
}
//当cur为空时,弹出栈顶元素(top),添加进顺序表,赋给cur其右子树
//若cur还是为空(top.right为空),循环过来继续弹出元素,并赋给cur新的top.right
TreeNode top = stack.pop();
list.add(top.val);
cur = top.right;
}
return list;
}
}
💎 7、题七:二叉树后序遍历递归实现与非递归实现
后序遍历的实现,在递归上,与前序、中序遍历的思想是相同的,在此不再赘述。
而在非递归的实现上,后序遍历需要额外注意几点。
🌟 7.1 递归实现->代码
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> list = new ArrayList<>();
if(root == null) {
return list;
}
List<Integer> list1 = postorderTraversal(root.left);
list.addAll(list1);
List<Integer> list2 = postorderTraversal(root.right);
list.addAll(list2);
list.add(root.val);
return list;
}
}
🔆 7.2 非递归实现->思路分析
后序遍历在非递归的实现上,需要额外注意几点:
- 当cur为空时,不可直接将栈顶元素弹出并添加进顺序表(或打印,只是访问根节点时的操作不同)
- 因为栈顶元素即根节点,后序遍历,需要判断栈顶元素的右孩子是否为空,若为空,才可弹出栈顶元素并打印;若右孩子不为空,需要先对右子树中的元素进行入栈和打印操作
- 而当栈顶元素的右孩子不为空时,我们又需要额外注意一点!!!当栈顶元素的右孩子不为空时,那就说明这个栈顶元素将永远不能出栈,会造成死循环!!!我们需要记录进过栈的元素,避免死循环的出现!!!
发生死循环过程如下图:
🔆 7.2.1 非递归实现->代码
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> list = new ArrayList<>();
Stack<TreeNode> stack = new Stack<>();
TreeNode cur = root;
//prev指向被打印过的节点
TreeNode prev = null;
//注意这里的循环条件
//因为后续cur为空时要弹出栈顶元素,故栈不能为空
//且一开始时栈为空,故cur不为空就可进入循环
//当栈空且cur为空,说明前序遍历已完成
while (cur != null || !stack.isEmpty()) {
//cur走左子树
while (cur != null) {
stack.push(cur);
cur = cur.left;
}
//当cur为空时,不可直接将栈顶元素弹出和添加进顺序表
//要判断栈顶元素(根节点)是否有右孩子
//若右孩子为空可以弹出和添加
//若右孩子不为空,需要先将右子树的元素入栈并添加
TreeNode top = stack.peek();
//当栈顶元素的右孩子入过栈时,此时栈顶元素可以弹出并添加进顺序表
if(top.right == null || top.right == prev) {
stack.pop();
list.add(top.val);
prev = top;
}else {
cur = top.right;
}
}
return list;
}
}
原文地址:https://blog.csdn.net/2401_83595513/article/details/140480592
免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!