二叉树层序遍历(LeetCode刷题!!!)
跟着carl哥的第十五天
二叉树层序遍历模板
class Solution {
public List<List<Integer>> resList = new ArrayList<List<Integer>>();
public List<List<Integer>> levelOrder(TreeNode root) {
//checkFun01(root,0);
checkFun02(root);
return resList;
}
//DFS--递归方式
public void checkFun01(TreeNode node, Integer deep) {
if (node == null) return;
deep++;
if (resList.size() < deep) {
//当层级增加时,list的Item也增加,利用list的索引值进行层级界定
List<Integer> item = new ArrayList<Integer>();
resList.add(item);
}
resList.get(deep - 1).add(node.val);
checkFun01(node.left, deep);
checkFun01(node.right, deep);
}
//BFS--迭代方式--借助队列
public void checkFun02(TreeNode node) {
if (node == null) return;
Queue<TreeNode> que = new LinkedList<TreeNode>();
que.offer(node);
while (!que.isEmpty()) {
List<Integer> itemList = new ArrayList<Integer>();
int len = que.size();
while (len > 0) {
TreeNode tmpNode = que.poll();
itemList.add(tmpNode.val);
if (tmpNode.left != null) que.offer(tmpNode.left);
if (tmpNode.right != null) que.offer(tmpNode.right);
len--;
}
resList.add(itemList);
}
}
}
一口气十道层序遍历题
102.二叉树的层序遍历
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
层序遍历一个二叉树。就是从左到右一层一层的去遍历二叉树。这种遍历的方式和我们之前讲过的都不太一样。
需要借用一个辅助数据结构即队列来实现,队列先进先出,符合一层一层遍历的逻辑,而是用栈先进后出适合模拟深度优先遍历也就是递归的逻辑。
而这种层序遍历方式就是图论中的广度优先遍历,只不过我们应用在二叉树上。
答案:
class Solution {
public List<List<Integer>> result = new ArrayList<>();
public List<List<Integer>> levelOrder(TreeNode root) {
checkFuntion(root);
return result;
}
private void checkFuntion(TreeNode root) {
if (root == null) {
return;
}
Queue<TreeNode> queue = new LinkedList<>();
TreeNode item;
queue.offer(root);
while (!queue.isEmpty()) {
//每一层的大小
int size = queue.size();
//每一层的结果集
List<Integer> res = new ArrayList<>();
for (int i = 0; i < size; i++) {
item = queue.poll();
res.add(item.val);
if (item.left != null) {
queue.add(item.left);
}
if (item.right != null) {
queue.add(item.right);
}
}
result.add(res);
}
}
}
107.二叉树的层次遍历II
给定一个二叉树,返回其节点值自底向上的层次遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)
思路:跟上一题一样,只是最后reverse一下结果集即可
答案:
class Solution {
public List<List<Integer>> result = new ArrayList<>();
public List<List<Integer>> levelOrderBottom(TreeNode root) {
checkFunction(root);
Collections.reverse(result);
return result;
}
private void checkFunction(TreeNode root) {
if (root == null) {
return;
}
Queue<TreeNode> queue = new LinkedList<>();
TreeNode item;
queue.offer(root);
while (!queue.isEmpty()) {
//每一层的大小
int size = queue.size();
//每一层的结果集
List<Integer> res = new ArrayList<>();
for (int i = 0; i < size; i++) {
item = queue.poll();
res.add(item.val);
if (item.left != null) {
queue.add(item.left);
}
if (item.right != null) {
queue.add(item.right);
}
}
result.add(res);
}
}
}
199.二叉树的右视图
给定一棵二叉树,想象自己站在它的右侧,按照从顶部到底部的顺序,返回从右侧所能看到的节点值。
思路:遍历到每一层最后一个元素再保存即可。
答案:
class Solution {
public List<Integer> res = new ArrayList<>();
public List<Integer> rightSideView(TreeNode root) {
checkFunction(root);
return res;
}
private void checkFunction(TreeNode root) {
if (root == null) {
return;
}
Queue<TreeNode> queue = new LinkedList<>();
int size = 0;
TreeNode item;
queue.add(root);
while(!queue.isEmpty()) {
size = queue.size();
while (size > 0) {
item = queue.poll();
if (size == 1) {
res.add(item.val);
}
if (item.left != null) {
queue.add(item.left);
}
if (item.right != null) {
queue.add(item.right);
}
size--;
}
}
}
}
637.二叉树的层平均值
给定一个非空二叉树, 返回一个由每层节点平均值组成的数组。
思路:跟上一题差不多,这个每一层用一个sum去加该层所有元素,然后再除以该层大小
答案:
class Solution {
public List<Double> res = new ArrayList<>();
public List<Double> averageOfLevels(TreeNode root) {
checkFunction(root);
return res;
}
private void checkFunction(TreeNode root) {
if (root == null) {
return;
}
Queue<TreeNode> queue = new LinkedList<>();
int size = 0;
int n = 0;
TreeNode item;
queue.add(root);
while(!queue.isEmpty()) {
size = n = queue.size();
double sum = 0.0;
while (size > 0) {
item = queue.poll();
sum += item.val;
if (item.left != null) {
queue.add(item.left);
}
if (item.right != null) {
queue.add(item.right);
}
size--;
}
res.add(sum / n);
}
}
}
429.N叉树的层序遍历
给定一个 N 叉树,返回其节点值的层序遍历。 (即从左到右,逐层遍历)。
返回其层序遍历:
[ [1], [3,2,4], [5,6] ]
思路:只是把两个孩子变成多个孩子,一样遍历即可
答案:
class Solution {
public List<List<Integer>> result = new ArrayList<>();
public List<List<Integer>> levelOrder(Node root) {
if (root == null) {
return result;
}
Queue<Node> queue = new LinkedList<>();
queue.add(root);
Node item = null;
int size = 0;
while (!queue.isEmpty()) {
size = queue.size();
List<Integer> res = new ArrayList<>();
for (int i = 0; i < size; i++) {
item = queue.poll();
res.add(item.val);
List<Node> children = item.children;
if (children == null || children.size() == 0) {
continue;
}
for (Node child : children) {
if (child != null) {
queue.add(child);
}
}
}
result.add(res);
}
return result;
}
}
515.在每个树行中找最大值
您需要在二叉树的每一行中找到最大的值。
思路:遍历每一层 存储该层最大值
答案:
class Solution {
public List<Integer> res = new ArrayList<Integer>();
public List<Integer> largestValues(TreeNode root) {
maxNum(root);
return res;
}
public void maxNum(TreeNode root) {
if (root == null)
return ;
Queue<TreeNode> que = new LinkedList<TreeNode>();
que.offer(root);
while(!que.isEmpty()){
int num = Integer.MIN_VALUE;;
int len = que.size();
while(len>0){
TreeNode node = que.poll();
if(node.val>=num)
num = node.val;
if(node.left!=null)
que.offer(node.left);
if(node.right!=null)
que.offer(node.right);
len--;
}
res.add(num);
}
}
}
116.填充每个节点的下一个右侧节点指针
给定一个完美二叉树,其所有叶子节点都在同一层,每个父节点都有两个子节点。二叉树定义如下:
填充它的每个 next 指针,让这个指针指向其下一个右侧节点。如果找不到下一个右侧节点,则将 next 指针设置为 NULL。
初始状态下,所有 next 指针都被设置为 NULL。
思路:遍历队列时,先拿出一个节点作为当前元素,再用peek()拿出下个元素的地址,再用当前元素的next指向下一元素即可/
答案:
class Solution {
public Node connect(Node root) {
if (root == null) {
return root;
}
Queue<Node> queue = new LinkedList<>();
int size = 0;
Node item = null;
queue.add(root);
while (!queue.isEmpty()) {
size = queue.size();
for (int i = 0; i < size; i++) {
item = queue.poll();
if (i < size - 1) {
item.next = queue.peek();
} else {
item.next = null;
}
if (item.left != null) {
queue.add(item.left);
}
if (item.right != null) {
queue.add(item.right);
}
}
}
return root;
}
}
117.填充每个节点的下一个右侧节点指针II
思路:跟上一次完全一样,代码可以直接AC
104.二叉树的最大深度
给定一个二叉树,找出其最大深度。
二叉树的深度为根节点到最远叶子节点的最长路径上的节点数。
说明: 叶子节点是指没有子节点的节点。
示例:
给定二叉树 [3,9,20,null,null,15,7],
思路:最大深度就是二叉树的层数。
答案:
class Solution {
public int maxDepth(TreeNode root) {
return getDepth(root);
}
private int getDepth(TreeNode root) {
if (root == null) {
return 0;
}
int leftDepth = getDepth(root.left);
int rightDepth = getDepth(root.right);
return Math.max(leftDepth, rightDepth) + 1;
}
}
111.二叉树的最小深度
给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明:叶子节点是指没有子节点的节点。
思路:需要注意的是,只有当左右孩子都为空的时候,才说明遍历的最低点了。如果其中一个孩子为空则不是最低点。
答案:
class Solution {
public int minDepth(TreeNode root) {
if (root == null) {
return 0;
}
Queue<TreeNode> queue = new LinkedList<>();
int size;
int count = 0;
TreeNode item;
queue.add(root);
while (!queue.isEmpty()) {
count++;
size = queue.size();
for (int i = 0; i < size; i++) {
item = queue.poll();
if (item.left != null) {
queue.add(item.left);
}
if (item.right != null) {
queue.add(item.right);
}
if (item.left == null && item.right == null) {
return count;
}
}
}
return count;
}
}
226.翻转二叉树
给你一棵二叉树的根节点 root ,翻转这棵二叉树,并返回其根节点。
思路:每一层都交换节点即可
答案
class Solution {
public TreeNode invertTree(TreeNode root) {
if(root==null)
return null;
Queue<TreeNode> que = new LinkedList<TreeNode>();
que.offer(root);
while(!que.isEmpty()){
int len = que.size();
while(len > 0) {
TreeNode node = que.poll();
swap(node);
if(node.left!=null)
que.offer(node.left);
if(node.right!=null)
que.offer(node.right);
len --;
}
}
return root;
}
public void swap(TreeNode node) {
TreeNode temp = node.left;
node.left = node.right;
node.right = temp;
}
}
101. 对称二叉树
给你一个二叉树的根节点 root , 检查它是否轴对称。
思路:每次判断左节点的左节点是否等于右节点的右节点,左节点的右节点是否等于右节点的左节点
则外侧是否相等,内侧是否相等。
答案:
class Solution {
public boolean isSymmetric(TreeNode root) {
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root.left);
queue.add(root.right);
while(!queue.isEmpty()) {
TreeNode leftNode = queue.poll();
TreeNode rightNode = queue.poll();
if (leftNode == null && rightNode == null) {
continue;
}
if (leftNode == null || rightNode == null || leftNode.val != rightNode.val) {
return false;
}
queue.add(leftNode.left);
queue.add(rightNode.right);
queue.add(leftNode.right);
queue.add(rightNode.left);
}
return true;
}
private boolean compare(TreeNode left, TreeNode right) {
if (left == null && right != null) {
return false;
}
if (left != null && right == null) {
return false;
}
if (left == null && right == null) {
return true;
}
if (left.val != right.val) {
return false;
}
//比较外侧
boolean compareOutside = compare(left.left, right.right);
//比较内侧
boolean compareInside = compare(left.right, right.left);
return compareInside && compareOutside;
}
}