Java [Leetcode 144]Binary Tree Preorder Traversal

Posted 2020-06-19 王小二的小博

题目描述：

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].

解题思路：

这个问题最简单的方法是使用递归，但是题目规定不能使用，得使用迭代的方法。

那么我们考虑使用栈来实现。

思路是每次遍历这节点，把该点的值放入list中，然后把该点的右孩子放入栈中，并将当前点设置为左孩子，然后迭代的去访问当前点；如果遇到当前点的左孩子为空，则弹出栈中的点（如果栈不为空的情况下），然后继续迭代。

代码如下：

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