原题链接:https://leetcode.com/problems/binary-tree-preorder-traversal/
1. 题目介绍
Given a binary tree, return the preorder traversal of its nodes’ values.
给出一个二叉树,返回它的前序遍历顺序。
前序遍历是指,首先访问根结点,然后遍历左子树,最后遍历右子树。
Example:
Follow up: Recursive solution is trivial, could you do it iteratively?
更进一步:递归的方法太简单了,可以用迭代做吗?
2. 解题思路
2.1 递归
递归的思路比较容易,使用深度优先搜索的方法,依次将根节点、根节点的左子树和根节点的右子树放入List中去。
实现代码
/**
* 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> l = new ArrayList<>();
return helper(root,l);
}
public List<Integer> helper(TreeNode root, List<Integer> l){
if(root == null){
return l;
}
l.add(root.val);
helper(root.left,l);
helper(root.right,l);
return l;
}
}
2.2 迭代
使用堆栈进行辅助。每遍历一个根节点的时候,首先把根节点的val值放入List中,然后将根节点的右子节点放入堆栈,然后继续遍历根节点的左子节点。当根节点的左子树为空时,就从堆栈顶部弹出当初保存的右子节点作为新的根节点进行遍历。
简单的说,就是先把自己的val放进去,然后一路向左遍历,同时把右边的都放到堆栈里面,当左边走不通的时候,就从堆栈里面取出一个右边的继续遍历。
当根节点为空并且堆栈也为空的时候,循环结束,便利完成。
实现代码
/**
* 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> l = new ArrayList<>();
Stack<TreeNode> s = new Stack<>();
TreeNode temp = root;
while(temp != null || s.isEmpty() != true){
if(temp != null){
l.add(temp.val);
if(temp.right != null){
s.push(temp.right);
}
temp = temp.left;
}else{
temp = s.pop();
}
}
return l;
}
}