题目描述
题目一
层序遍历即可
class Solution {
/**
* 广度优先遍历
*
* @param root
* @return
*/
public int[] levelOrder(TreeNode root) {
if (root == null) {
return new int[0];
}
List<Integer> list = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while (!queue.isEmpty()) {
int len = queue.size();
while (len > 0) {
TreeNode node = queue.poll();
list.add(node.val);
if (node.left != null) queue.offer(node.left);
if (node.right != null) queue.offer(node.right);
len--;
}
}
int[] res = new int[list.size()];
for (int i = 0; i < res.length; i++) {
res[i] = list.get(i);
}
return res;
}
}
题目二
也是层序遍历
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
if (root == null) {
return new ArrayList<>();
}
List<List<Integer>> list = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
while (!queue.isEmpty()) {
int len = queue.size();
List<Integer> list1 = new ArrayList<>();
while (len > 0) {
TreeNode node = queue.poll();
list1.add(node.val);
if (node.left != null) queue.offer(node.left);
if (node.right != null) queue.offer(node.right);
len--;
}
list.add(list1);
}
return list;
}
}
题目三
层序遍历 + 双端队列:
利用双端队列的两端皆可添加元素的特性,设打印列表(双端队列) tmp ,并规定:
- 奇数层 则添加至 tmp 尾部 ,
偶数层 则添加至 tmp 头部 。
/**
* 双端队列
*
* @param root
* @return
*/
public List<List<Integer>> levelOrder(TreeNode root) {
if (root == null) {
return new ArrayList<>();
}
List<List<Integer>> list = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
int count = 1;
queue.add(root);
while (!queue.isEmpty()) {
int len = queue.size();
LinkedList<Integer> list1 = new LinkedList<>();
while (len > 0) {
TreeNode node = queue.poll();
if (count % 2 == 0) {
list1.addLast(node.val);
}else list1.addFirst(node.val);
if (node.left != null) queue.offer(node.left);
if (node.right != null) queue.offer(node.right);
len--;
}
count++;
list.add(list1);
}
return list;
}
层序遍历 + 倒序
```java /**
- 反转解决 *
- @param root
- @return
*/
public List
- > levelOrder(TreeNode root) {
if (root == null) {
return new ArrayList<>();
}
List
- > list = new ArrayList<>();
Queue
queue = new LinkedList<>(); int count = 1; queue.add(root); while (!queue.isEmpty()) { int len = queue.size(); List list1 = new ArrayList<>(); while (len > 0) { TreeNode node = queue.poll();
list1.add(node.val);
if (node.left != null) queue.offer(node.left);
if (node.right != null) queue.offer(node.right);
len--;
} count++; // 表示奇数行还是偶数行 if (count % 2 == 1) Collections.reverse(list1); list.add(list1); } return list; }
```