给定一个二叉树,返回它的 前序 遍历。
示例:
输入: [1,null,2,3]
1
\
2
/
3
输出: [1,2,3]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
解题思路:用栈实现,
1.打印当前节点
2.若存在右子树,则右子树入栈
3.若存在左子树,则指向左子树,否则,栈不空则出栈,栈空则结束
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> list = new ArrayList<>();
if(root==null)
return list;
Stack<TreeNode> stack = new Stack<>();
TreeNode p = root;
while(p!=null){
list.add(p.val);
if(p.right!=null){
stack.push(p.right);
}
if(p.left!=null){
p = p.left;
}else{
if(stack.isEmpty()){
break;
}else {
p = stack.pop();
}
}
}
return list;
}
}