题目一
94. 二叉树的中序遍历
给定一个二叉树的根节点 root ,返回它的 中序 遍历。
示例 1:
输入:root = [1,null,2,3]
输出:[1,3,2]
示例 2:
输入:root = []
输出:[]
示例 3:
输入:root = [1]
输出:[1]
示例 4:
输入:root = [1,2]
输出:[2,1]
示例 5:
输入:root = [1,null,2]
输出:[1,2]
提示:
树中节点数目在范围 [0, 100] 内
-100 <= Node.val <= 100
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
我的答案
递归
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> ret = new ArrayList<>();
if(root == null) return ret;
getValue(root, ret);
return ret;
}
public void getValue(TreeNode root, List<Integer> ret){
if(root != null){
getValue(root.left, ret);
ret.add(root.val);
getValue(root.right, ret);
}
}
}
迭代
// 迭代
class Solution {
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> ret = new ArrayList<>();
if(root == null) return ret;
Stack<TreeNode> stack = new Stack<TreeNode>();
TreeNode node = root;
while(!stack.isEmpty() || node != null){
while(node != null){
stack.push(node); // 先记录前一个node状态
node = node.left; // 去往左节点
}
node = stack.pop(); // 取回前一个node状态
ret.add(node.val); // 添加值,这时已经没有左节点了,就把到这的当前节点的值记录下来
node = node.right; // 再去往右节点
}
return ret;
}
}
题目二
145. 二叉树的后序遍历
给定一个二叉树,返回它的 后序 遍历。
示例:
输入: [1,null,2,3]
1
2
/
3
输出: [3,2,1]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
我的答案
遍历
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> ret = new ArrayList<>();
if(root == null) return ret;
getValue(root, ret);
return ret;
}
public void getValue(TreeNode root, List<Integer> ret){
if(root != null){
getValue(root.left, ret);
getValue(root.right, ret);
ret.add(root.val); // 最后再添加值进去
}
}
}
迭代
// 迭代
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> ret = new ArrayList<>();
if(root == null) return ret;
Stack<TreeNode> stack = new Stack<TreeNode>();
TreeNode preNode = null; // 用于记录上一个节点的状态的
while(!stack.isEmpty() || root != null){
while(root != null){
stack.push(root); // 先记录前一个node状态
root = root.left; // 去往左节点
}
root = stack.pop(); // 取回前一个node状态
if(root.right == null || root.right == preNode){
// 右节点为空,或者右节点已经被遍历过了
ret.add(root.val); // 就把当前的节点的值添加进列表中
preNode = root; // 记录当前已添加的节点
root = null; // 将root置为null,防止一直添加
}else{
// 右节点不为null,就添加进栈
stack.push(root);
root = root.right; // 去往右节点
}
}
return ret;
}
}
题目三
102. 二叉树的层序遍历
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
我的答案
我没写出来
广度优先搜索(BFS)
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> ret = new ArrayList<List<Integer>>();
if (root == null) return ret;
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.offer(root); // 先将根节点入队
while(!queue.isEmpty()){ // 队列不空就继续执行
List<Integer> layer = new ArrayList<>(); // 每层列表
int queueSize = queue.size(); // 每层的队列长度
for(int i = 1; i <= queueSize; i++){
TreeNode node = queue.poll(); // poll移除并返回队列的头数据
layer.add(node.val);
if(node.left != null){
queue.offer(node.left); // 左节点入队
}
if(node.right != null){
queue.offer(node.right); // 右节点入队
}
}
ret.add(layer); // 把每层的数据添加进去
}
return ret;
}
}
复杂度分析
记树上所有节点的个数为 n。
时间复杂度:每个点进队出队各一次,故渐进时间复杂度为 O(n)。
空间复杂度:队列中元素的个数不超过 n 个,故渐进空间复杂度为 O(n)。