题目:
给定一个二叉树根节点,返回二叉树的中序遍历。
递归方法实现
迭代方法实现
示例1:
1
\
2
/
3
输入:root = [1,null,2,3]
输出:[1,3,2]
示例2:
输入:root = []
输出:[]
思路:
这是一道简单题,就是二叉树的中序遍历,中序遍历就是左子树——>根节点——>右子树的遍历顺序。不过注意示例二,如果二叉树为空返回的不是null。
违背开发习惯的递归代码:
class Solution {
//定义一个链表接受数据
List<Integer> list = new ArrayList<>();
public List<Integer> inorderTraversal(TreeNode root) {
//边界条件判断
if (root == null) {
return new ArrayList();
}
//先遍历左子树
inorderTraversal(root.left);
//将根节点值放入链表
list.add(root.val);
//再遍历右子树
inorderTraversal(root.right);
return list;
}
}
符合开发习惯的代码:
class Solution {
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<Integer>();
inorder(root, res);
return res;
}
public void inorder(TreeNode root, List<Integer> res) {
if (root == null) {
return;
}
inorder(root.left, res);
res.add(root.val);
inorder(root.right, res);
}
}
迭代方式实现中序遍历:
中序遍历迭代方式实现逻辑如下,很巧妙的逻辑。自己画个二叉树跟着代码debug一遍就可以理解。
class Solution {
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<Integer>();
Deque<TreeNode> stk = new LinkedList<TreeNode>();
while (root != null || !stk.isEmpty()) {
while (root != null) {
stk.push(root);
root = root.left;
}
root = stk.pop();
res.add(root.val);
root = root.right;
}
return res;
}
}