二叉树的前序遍历

发表于
本文字数： 1.6k 阅读时长 ≈ 1 分钟

LeetCode每日一题，144. Binary Tree Preorder Traversal

先看题目描述

BQmsmR.png

大意就是给定一个二叉树，让我们返回其先序遍历序列

算法和思路

递归

直接用递归的算法十分简单，定义 dfs(root, res) 表示当前遍历到 root 节点的答案，那么按照定义，我们只要递归调用 dfs(root.left, res) 来遍历root 节点的左子树，然后将 root 节点的值加入答案，再递归调用 dfs(root.right, res) 来遍历 root 节点的右子树即可，递归终止的条件为碰到空节点

栈

其实就是上一方法中的递归函数用迭代来实现，两种方式等价。区别在于递归的时候维护了一个隐形的栈，而在迭代的时候则是显式的将这个栈模拟出来

算法源码

递归

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
import java.util.*;

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        dfs(root, res);
        return res;
    }

    private void dfs(TreeNode root, List<Integer> res) {
        if (root == null) return;
        res.add(root.val);
        dfs(root.left, res);
        dfs(root.right, res);
    }
}

栈

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        Deque<TreeNode> stack = new LinkedList<TreeNode>();
        List<Integer> res = new ArrayList<>();
        if (root == null) return res;
        stack.add(root);
        while (!stack.isEmpty()) {
            root = stack.removeLast();
            res.add(root.val);
            if (root.right != null) {
                stack.add(root.right);
            }
            if (root.left != null) {
                stack.add(root.left);
            }
        }
        return res;
    }
}