Leetcode144.二叉树前序遍历 Binary Tree Preorder Traversal(Java)

144.Binary Tree Preorder Traversal(Java)

##Stack##, ##Tree##

题解

二叉树的前序遍历，即遍历顺序为根=>左=>右

递归做法较为简洁

采用非递归做法时要借助栈，对于每个当前结点的右儿子(非空)要最先进栈，因为进栈顺序要取遍历顺序的相反序

**时间复杂度：**O(n)

递归

class Solution {
    LinkedList<Integer> res = new LinkedList<>();
    
    public List<Integer> preorderTraversal(TreeNode root) {
        traversal(root);
        return res;
        
    }
    
    private void traversal(TreeNode root) {
        if (root == null) return;
        res.add(root.val);
        traversal(root.left);
        traversal(root.right);
    }
}

---

非通用迭代

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        Stack<TreeNode> sta = new Stack<>();
        LinkedList<Integer> res = new LinkedList<>();
        if (root == null) return res;
        sta.push(root);
        while (!sta.isEmpty()) {
            TreeNode curr = sta.pop();
            res.add(curr.val);
            if (curr.right != null) sta.push(curr.right);
            if (curr.left != null) sta.push(curr.left);
        }
        return res;
    }
}

---

通用迭代法

通用迭代法指该迭代思路在树的前中后序遍历种都可以使用
二叉树中序遍历(通用迭代法)
二叉树后序遍历(通用迭代法)

二叉树非递归遍历需要解决的问题：遍历结点和将结点加入到解集合不同步的问题。(以中序遍历为例，遍历到一个结点时，该节点的左子树部分比该结点更早加入到解集合种)

解决方法为：可以将访问的结点都放入栈中，但是将需要处理的结点做标记，标记的方法为在需要处理的结点放入栈后，紧接着放一个空指针作为标记

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        Stack<TreeNode> sta = new Stack<>();
        ArrayList<Integer> res = new ArrayList<>();
        
        if (root == null) return res;
        sta.push(root);
        while (!sta.isEmpty()) {
            TreeNode curr = sta.pop();
            if (curr != null) {
                if (curr.right != null) sta.push(curr.right);
                if (curr.left != null) sta.push(curr.left);
                sta.push(curr);
                sta.push(null);
            } else {
                curr = sta.pop();
                res.add(curr.val);
            }
        }
        return res;
    }
}