先序遍历二叉树
定义节点
class TreeNode{
int val;
TreeNode left;
TreeNode right;
public TreeNode(){}
public TreeNode(int val){
this.val = val;
}
public 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> list = new LinkedList<>();
traversal(root,list);
return list;
}
public void traversal(TreeNode root,List list){
//递归结束条件
if(root==null){
return;
}
//先序遍历的顺序为中左右
list.add(root.val);
traversal(root.left,list);
traversal(root.right,list);
}
}
非递归实现:
首先利用栈存储节点,第一次把根节点存入栈中,进入循环,将栈顶元素出栈并存储在list中,接着将先前出栈的栈顶元素的右节点进栈,然后左节点进栈,当下一次循环,左节点出栈时,保存该节点到list中,然后将该节点的右节点进栈,左节点再进栈,直到栈为空时,循环结束。
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
Stack<TreeNode> stack = new Stack<>();
ArrayList<Integer> list = new ArrayList<>();
if((root==null)){
return list;
}
stack.push(root);
while(!stack.isEmpty()){
TreeNode tmproot = stack.pop();
list.add(tmproot.val);
if(tmproot.right!=null){
stack.push(tmproot.right);
}
if(tmproot.left!=null){
stack.push(tmproot.left);
}
}
return list;
}
}