LeetCode題144. 二叉樹的前序遍歷

這裡真安靜發表於2020-10-27
題目描述:
	給定一個二叉樹,返回它的 前序 遍歷。
例子:
	輸入: [1,null,2,3]  
   1
    \
     2
    /
   3 

輸出: [1,2,3]
進階: 遞迴演算法很簡單,你可以通過迭代演算法完成嗎?
思路:
	1.遞迴
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    private List<Integer> list = new ArrayList() ;
    public List<Integer> preorderTraversal(TreeNode root) {
        if(root == null){
            return new ArrayList() ;
        }
        list.add(root.val) ;
        if(root.left != null){
            preorderTraversal(root.left);
        }
        if(root.right != null){
            preorderTraversal(root.right);
        }
        return list ;
    }
}
2.迭代
	使用棧,沒存入一個節點的值到list集合中,就將這個節點入棧,並且向左遍歷,直到沒有左節點。這時就從棧中取出一個節點(由於棧有先進後出的特點,所以取出的一定是當前左節點的父節點),得到取出節點的右節點。
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> list = new ArrayList() ;
        if(root == null){
            return list ;
        }
        Stack<TreeNode> stack = new Stack<TreeNode>();
        TreeNode node = root ;
        while(!stack.isEmpty() || node != null){
            while(node != null){
                list.add(node.val);
                stack.push(node);
                node = node.left ;
            }
            node = stack.pop() ;
            node = node.right ;
        }
        return list ;
    }
}

相關文章