1. 程式人生 > >LeetCode144. 二叉樹的前序遍歷

LeetCode144. 二叉樹的前序遍歷

給定一個二叉樹,返回它的 前序 遍歷。

 示例:

輸入: [1,null,2,3]  
   1
    \
     2
    /
   3 

輸出: [1,2,3]

進階: 遞迴演算法很簡單,你可以通過迭代演算法完成嗎?

思路:

遞迴實現二叉樹的前序遍歷。

/**
 * 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 LinkedList<Integer>();
        preorderTraversal(root,res);
        return res;
    }
     public  void preorderTraversal(TreeNode root,List<Integer> res){
        if(root==null){
            return ;
        }
        res.add(root.val);
        if(root.left!=null){
            preorderTraversal(root.left,res);
        }
        if(root.right!=null){
            preorderTraversal(root.right,res);
        }
    }
}

非遞迴實現,藉助stack結構。

/**
 * 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 LinkedList<Integer>();
         if(root==null){
            return res;
        }
        ArrayDeque<TreeNode> stack=new ArrayDeque<TreeNode>();
        stack.push(root);
        while(!stack.isEmpty()){
            TreeNode node=stack.pop();
            res.add(node.val);
            if(node.right!=null){
                stack.push(node.right);
            }
            if(node.left!=null){
                stack.push(node.left);
            }
        }
        return res;
    }
    
}