Leetcode binary-tree-preorder- traversal 二叉树的前序遍历 Java

题目描述

 

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

For example:
Given binary tree{1,#,2,3},

   1
    \
     2
    /
   3

 

return[1,2,3].

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

递归方法是无价值的，能不能使用迭代来完成

解法一：递归>_<

/**
     * 递归实现
     * 根据先序遍历的特点，根->左->右
     * @param root
     * @return
     */
    public List<Integer> preorderTraversalRecursive(TreeNode root) {
        List<Integer> list = new ArrayList<>();
 
        //判断是否为空树
        if (root == null) {
            return list;
        }
 
        list.add(root.val); //【根】
        list.addAll(preorderTraversalRecursive(root.left)); //【左】通过addAll方法，将根节点的左子树，按preorderTraversalRecursive(root.left)返回的顺序，添加到列表尾部
        list.addAll(preorderTraversalRecursive(root.right));//【右】通过addAll方法，将根节点的右子树，按preorderTraversalRecursive(root.right)返回的顺序，添加到列表尾部
 
        return list;
    }

解法二：迭代

使用栈来保存root，然后取出，经结点的值加入到arraylist中，接着分别将结点的右孩子跟左孩子压入栈中，直到访问了所有的结点

代码如下：

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
import java.util.ArrayList;
import java.util.Stack;
public class Solution {
    public ArrayList<Integer> preorderTraversal(TreeNode root) {
       ArrayList<Integer> result = new ArrayList<Integer>();
        if(root == null)
            return result;
        Stack<TreeNode> stack = new Stack();
        stack.push(root);
        while(!stack.isEmpty())
        {
            TreeNode node = stack.pop();
            result.add(node.val);
            if(node.right!=null)
                stack.push(node.right);
            if(node.left != null)
                stack.push(node.left);
        }
        return result;
    }
}