Leetcode 144 - 二叉树的前序遍历

题解

方法一：递归法

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        preOrder(root,list);
        return list;
       
    }
    
    //为了递归返回，递归函数的返回值需要为null，所以这里不能用原有的函数，需要新起一个函数
    //为了传递list的值，需要把list加入函数参数里面
    public void preOrder(TreeNode root,List<Integer> list){
        //递归边界
        if(root==null) return;
        //递归在本节点应做的事：加入本节点值
        list.add(root.val);
        //递归左子树
        preOrder(root.left,list);
        //递归右子树
        preOrder(root.right,list);
    }
}

方法二：迭代法
题解转载自gre-z
首先我们应该创建一个Stack用来存放节点，首先我们想要打印根节点的数据，此时Stack里面的内容为空，所以我们优先将头结点加入Stack，然后打印。

之后我们应该先打印左子树，然后右子树。所以先加入Stack的就是右子树，然后左子树。
此时你能得到的流程如下:

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