题目描述:
给定一个二叉树,返回它的前序遍历。
分析:
递归算法与非递归算法。在非递归的方法中,使用辅助栈Stack<TreeNode>,右子树先入栈左子树后入栈,就是根、左、右的顺序。
代码:
/**
* 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> res = new ArrayList();
if(root==null){
return res;
}
res.add(root.val); //非空的情况
preOrder(root.left,res);
preOrder(root.right,res);
return res;
}
public void preOrder(TreeNode root,List<Integer> list){
if(root==null){
return; //为空则停止
}
list.add(root.val); //先将该节点的值加入list,再判断其左右子树
preOrder(root.left,list);
preOrder(root.right,list);
}
}
//非递归算法 利用辅助栈 先入后出
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList();
if(root==null){
return res;
}
Stack<TreeNode> stack = new Stack(); //利用栈递归
stack.push(root);
while(!stack.isEmpty()){
TreeNode temp = stack.pop();
res.add(temp.val);
if(temp.right!=null) stack.push(temp.right);
if(temp.left!=null) stack.push(temp.left);
}
return res;
}
}