1、递归法
其实递归法提交结果也挺好看的,代码如下:
class Solution {
//前序遍历
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<Integer>();
preorder(root, res);
return res;
}
public void preorder(TreeNode root, List<Integer> res) {
if (root == null) {
return;
}
res.add(root.val);
preorder(root.left, res);
preorder(root.right, res);
}
}
Python3 (该代码源自大佬:zhywanna - 力扣(LeetCode)):
class Solution:
def preorderTraversal(self, root: Optional[TreeNode]) -> List[int]:
if not root:
return []
# python中 * 号是解包符号,可以解除数据结构的格式直接得到内部元素
return [root.val, *self.preorderTraversal(root.left), *self.preorderTraversal(root.right)]
2、迭代法
class Solution {
//前序遍历
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<Integer>();
if (root == null) {
return res;
}
//采用栈的数据结构
Deque<TreeNode> stack = new LinkedList<TreeNode>();
TreeNode node = root;
while(!stack.isEmpty() || node != null){
while (node != null) {
res.add(node.val);
stack.push(node); //添加到栈顶
node = node.left;
}
node = stack.pop(); //从栈顶删除
node = node.right;
}
return res;
}
}
3、完整题目
给你二叉树的根节点 root
,返回它节点值的 前序 遍历。
示例 1:
输入:root = [1,null,2,3] 输出:[1,2,3]
示例 2:
输入:root = [] 输出:[]
示例 3:
输入:root = [1] 输出:[1]
示例 4:
输入:root = [1,2] 输出:[1,2]
示例 5:
输入:root = [1,null,2] 输出:[1,2]
提示:
- 树中节点数目在范围
[0, 100]
内 -100 <= Node.val <= 100
进阶:递归算法很简单,你可以通过迭代算法完成吗?