二叉树的遍历144.94.145(递归和迭代)

322 阅读1分钟

迭代(利用栈实现)

前序遍历为例,遍历顺序中->左->右,栈特点为先进后出。
所以放入顺序为中->取出中->右->左,这样左在下一步就成了中,以此类推。

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        if (root == null){
            return result;
        }
        Stack<TreeNode> stack = new Stack<>();
        //入栈
        stack.push(root);
        while (!stack.isEmpty()){
        //从栈取出并放入结果集
            TreeNode node = stack.pop();
            result.add(node.val);
            if (node.right != null){
                stack.push(node.right);
            }
            if (node.left != null){
                stack.push(node.left);
            }
        }
        return result;
    }
}

递归

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<Integer>();
        preorder(root, result);
        return result;
    }

    public void preorder(TreeNode root, List<Integer> result) {
        if (root == null) {
            return;
        }
        result.add(root.val);//中
        preorder(root.left, result);//左
        preorder(root.right, result);//右
    }
}