Java [Leetcode 144]Binary Tree Preorder Traversal

时间:2021-01-13 15:07:38

题目描述:

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;
}
}