1. 程式人生 > >LeetCode-Binary Tree Preorder Traversal

LeetCode-Binary Tree Preorder Traversal

一、Description

Given a binary tree, return the preorder traversal of its nodes' values.

題目大意:輸出一個二叉樹的前序遍歷序列。

Example:

Input: [1,null,2,3]
   1
    \
     2
    /
   3

Output: [1,2,3]

二、Analyzation

兩種思路:

  1. 通過前序遞迴,每次訪問一個結點,把該結點的值加到list中,再訪問它的左結點和右結點。
  2. 通過一個棧來將結點儲存並以此彈出並訪問,由於list中的順序是先左後右,因此棧的加入是先右後左。

三、Accepted code

recursive version:

class Solution {
    List<Integer> list = new ArrayList<>();
    public List<Integer> preorderTraversal(TreeNode root) {
        if (root == null) {
            return list;
        }
        help(root);
        return list;
    }
    public void help(TreeNode root) {
        list.add(root.val);
        if (root.left != null) {
            help(root.left);
        }
        if (root.right != null) {
            help(root.right);
        }
    }
}

iterative version:

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        if (root == null) {
            return list;
        }
        Stack<TreeNode> stack = new Stack<>();
        stack.add(root);
        while (!stack.isEmpty()) {
            TreeNode temp = stack.pop();
            list.add(temp.val);
            if (temp.right != null) {
                stack.add(temp.right);
            }
            if (temp.left != null) {
                stack.add(temp.left);
            }
        }
        return list;
    }
}