Binary Tree Preorder Traversal

  • 描述
    Given a binary tree, return the preorder traversal of its nodes’ values. For example: Given binary tree
    {1, #, 2, 3}, return [1, 2, 3].

Note: Recursive solution is trivial, could you do it iteratively?

  • 使用非遞歸進行先序遍歷,要借助棧(后進先出),先把左子樹依次進棧,存入到結果集,這時候結果集保留了左子樹的節點值,接著彈棧,如果彈出的節點有右子樹,則把右子樹進棧,并保存到結果集

  • 時間復雜度O(n),空間復雜度O(n)

  • 中序遍歷只需要把輸出結果的位置調換一下即可



import java.util.ArrayList;
import java.util.List;
import java.util.Stack;

public class Solution {
    public List<Integer> inorderTraverse(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        if(root == null)
            return result;
        Stack<TreeNode> stack = new Stack<>();
        TreeNode p = root;
        while(p != null || !stack.isEmpty()) {
            if(p != null) {
                stack.push(p);
                result.add(p.val);
                p = p.left;
            }
            else {
                p = stack.pop();
                p = p.right;
            }
        }
        return result;
    }
}

import java.util.ArrayList;
import java.util.List;
import java.util.Stack;

// 解法2, 深度優先搜索DFS的思想
public class Solution1 {
    public static List<Integer> preorderTraverse(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        if(root == null)
            return result;
        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        while(!stack.isEmpty()) {
            TreeNode p = stack.pop();
            result.add(p.val);
            if(p.right != null) stack.push(p.right);
            if(p.left != null)  stack.push(p.left);
        }
        return result;
    }
}

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容