给定一个二叉树的根节点 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)
链接:https://leetcode-cn.com/problems/binary-tree-inorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
迭代算法,运用栈来进行解答,首先创建一个链表和一个栈,将根节点入栈,当栈为不为空或者节点为不为空的时候,就继续循环,如果都为空,说明整个树都已经运算完毕,直接跳出循环,如果此时的节点不是空,根据中序遍历的 左根右 进行分析,将左子树入栈,继续迭代,如果此时的节点是空的,说明左子树已经没了,此时让 节点tree = 此时的栈顶元素,出栈,然后添加到链表中即可,以下是示意图
步骤:
1、根节点1入栈
2、左孩子2入栈
3、左孩子4入栈
4、节点为空,将节点指向栈顶元素,左孩子4出栈
5、节点为空,将节点指向栈顶元素,节点2出栈
6、右孩子5入栈
7、右孩子5出栈
8、节点为空,节点指向栈顶元素1, 根节点1出栈
9、略。。。
/**
* 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 {
LinkedList<Integer> list = new LinkedList<>();
public List<Integer> inorderTraversal(TreeNode root) {
if(root == null) return list;
Stack<TreeNode> st = new Stack<TreeNode>();
TreeNode tree = root;
while(tree != null || !st.isEmpty()) {
if(tree != null) {
st.push(tree);
tree = tree.left;
} else {
tree = st.pop();
list.add(tree.val);
tree = tree.right;
}
}
return list;
}
}