乐趣区

关于leetcode:leetcode栈之二叉树的前序遍历

本文次要记录一下 leetcode 栈之二叉树的前序遍历

题目

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

输出: [1,null,2,3]  
   1
    \
     2
    /
   3 

输入: [1,2,3]

进阶: 递归算法很简略,你能够通过迭代算法实现吗?起源:力扣(LeetCode)链接:https://leetcode-cn.com/problems/binary-tree-preorder-traversal
著作权归领扣网络所有。商业转载请分割官网受权,非商业转载请注明出处。

题解

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) {this.val = val;}
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {public List<Integer> preorderTraversal(TreeNode root) {List<Integer> result = new ArrayList<>();
        preTraverse(root, result);
        return result;
    }

    private void preTraverse(TreeNode node, List<Integer> result) {if (node == null) {return ;}
        Stack<TreeNode> stack = new Stack<>();
        stack.push(node);
        while (!stack.empty()) {TreeNode treeNode = stack.pop();
            result.add(treeNode.val);
            if (treeNode.right != null) {stack.push(treeNode.right);
            }
            if (treeNode.left != null) {stack.push(treeNode.left);
            }
        }
    }
}

小结

这里借助栈来实现二叉树的前序遍历;在每个 preTraverse 办法外头新建一个栈,push 以后 node,而后循环 pop 栈,将元素增加到 result 中,之后先 push 右子节点,再 push 左子节点。

doc

  • 二叉树的前序遍历
退出移动版