LeetCode笔记：94. Binary Tree Inorder Traversal

Cloudox

发布于 2021-11-23 14:09:00

1910

发布于 2021-11-23 14:09:00

文章被收录于专栏：月亮与二进制月亮与二进制

问题：

Given a binary tree, return the inorder traversal of its nodes' values. For example: Given binary tree [1,null,2,3],

image.png return [1,3,2]. Note: Recursive solution is trivial, could you do it iteratively?

大意：

给出一个二叉树，返回中序遍历的节点值。比如：给出二叉树 [1,null,2,3]，

image.png 返回 [1,3,2]。注意：递归的解决方法很简单，能不能用循环做？

思路：

所谓中序遍历是指：左中右，这种遍历方式。

对于二叉树，因为要遍历，我们需要记录节点，否则从子节点是无法回到父节点的，所以我们需要使用栈，同时利用其先入后出的特性。

因为要不停地看一个节点的子节点然后回来，又要满足中序遍历，我们用递归来保证深入到叶子节点后能一个个返回来。

因为栈是先入后出的，而我们要记录的顺序是左中右，所以入栈的顺序应该反过来，即右中左，先入右节点，没有右节点了才入根节点，然后对左节点进行同样的操作。

全部遍历完后再一个个出栈记录节点值就可以了。

代码（Java）：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<Integer>();
        if (root == null) return result;
        
        Stack<TreeNode> stack = new Stack<TreeNode>();
        TreeNode node = root;
        findNode(node, stack);
        
        while (!stack.isEmpty()) {
            TreeNode temp = stack.pop();
            result.add(temp.val);
        }
        return result;
    }
    
    public void findNode(TreeNode node, Stack<TreeNode> stack) {
        if (node.right != null) {
            findNode(node.right, stack);
            stack.push(node);
        } else {
            stack.push(node);
        }
        if (node.left != null) {
            findNode(node.left, stack);
        }
    }
}

他山之石：

public List<Integer> inorderTraversal(TreeNode root) {
    List<Integer> list = new ArrayList<Integer>();

    Stack<TreeNode> stack = new Stack<TreeNode>();
    TreeNode cur = root;

    while(cur!=null || !stack.empty()){
        while(cur!=null){
            stack.add(cur);
            cur = cur.left;
        }
        cur = stack.pop();
        list.add(cur.val);
        cur = cur.right;
    }

    return list;
}

这个做法就是不用递归而用循环了，也是用栈，一路入栈左节点到底，然后出栈取值，这时候其实是一个没有左子节点的根节点了，然后对其右节点进行同样的操作，弄完了就返回上一个节点，其实也是左中右的顺序。

合集：https://github.com/Cloudox/LeetCode-Record

查看作者首页

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2017/11/23 ，如有侵权请联系 cloudcommunity@tencent.com 删除

二叉树