https://oj.leetcode.com/problems/binary-tree-preorder-traversal/
Given a binary tree, return the preorder traversal of its nodes' values.
For example:
Given binary tree {1,#,2,3}
,
1 \ 2 / 3
return [1,2,3]
.
Note: Recursive solution is trivial, could you do it iteratively?
public List<Integer> preorderTraversal(TreeNode root)
这一题的recursion做法其实很简单的。只要记得前序是中左右,中序是左中右,后序是左右中即可。
给出recursion代码
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> res =new LinkedList<Integer>();
helper(res, root);
return res;
}
public void helper(List<Integer> res, TreeNode root){
if(root != null){
res.add(root.val);
helper(res, root.left);
helper(res, root.right);
}
}
中序
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> res =new LinkedList<Integer>();
helper(res, root);
return res;
}
public void helper(List<Integer> res, TreeNode root){
if(root != null){
helper(res, root.left);
<pre name="code" class="java"> res.add(root.val);
helper(res, root.right); } }
后序
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> res =new LinkedList<Integer>();
helper(res, root);
return res;
}
public void helper(List<Integer> res, TreeNode root){
if(root != null){
helper(res, root.left);
helper(res, root.right);
res.add(root.val);
}
}
看见没,中左右,左中右,左右中。左和右都是递归,中是输出。。
关于循环的做法,其实这是一个很经典的用循环加Stack模拟递归的做法。理论上所有的递归都可以利用Stack来进行模拟。可以认为是这里是用状态stack来替代了call stack。这里可以简单的做出来只是因为这里递归所继承的状态量很简单,递归结构也很单纯。如果递归的状态量很大,递归结构很复杂,这个状态stack维护起来就变得有点麻烦了。
理论上还是左中右。遇到中间点就输出数据,然后先放入左节点进入状态stack,然后当左状态走到底就放右状态的stack。给出代码如下:
public List<Integer> preorderTraversal(TreeNode root) {
Stack<TreeNode> stack = new Stack<TreeNode>();
List<Integer> res = new LinkedList<Integer>();
while(root != null || !stack.isEmpty()){
if(root != null){
res.add(root.val);
stack.push(root);
root = root.left;
}else{
root = stack.pop();
root = root.right;
}
}
return res;
}