LeetCode144:二叉树的前序遍历

给定一个二叉树,返回它的 前序 遍历。

示例:
输入: [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;
    }
}

猜你喜欢

转载自blog.csdn.net/u012485480/article/details/80775719