标题 | 94. 二叉树的中序遍历 |
难度 | 简单 |
天数 | 第10天,第2/3题 |
数据结构 | 树 |
给定一个二叉树的根节点 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
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
以上内容来源:力扣(LeetCode)
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
整体思路参考《力扣-数据结构】【第 10 天】144. 二叉树的前序遍历》
- 中序遍历 先左,后根,再右
- 我们先将右节点添加进栈中
if(node.right != null){
stack.push(node.right);
}
- 然后处理根节点和左节点
- 左节点存在
- 左节点不存在
1.左节点存在
- 先将根节点存入栈中,然后存一个null,用于标记,表示遇到null需要记录根节点
if(node.left != null){
//如果左节点不为空,那么先添加一下当前更节点
stack.push(node);
//再添加一个标记
stack.push(null);
stack.push(node.left);
}
遇到null,记录根节点
if(node == null){
list.add(stack.pop().val);
}
- 左节点不存在,直接将根节点添加到list中
list.add(node.val);
完整代码:
/**
* 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 {
//数据结构 第 10 天 2/3 树
//中序遍历 先左,后根,再右
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> list = new ArrayList();
if(root == null){
return list;
}
Stack<TreeNode> stack = new Stack();
stack.push(root);
while(!stack.isEmpty()){
TreeNode node = stack.pop();
if(node != null){
//不是标记,正常添加左右子树到栈中
if(node.right != null){
stack.push(node.right);
}
if(node.left != null){
//如果左节点不为空,那么先添加一下当前更节点
stack.push(node);
//再添加一个标记
stack.push(null);
stack.push(node.left);
}else{
list.add(node.val);
}
}else{
//node == null 拿到标记,代表需要添加根节点值
list.add(stack.pop().val);
}
}
return list;
}
}