递归形式
public List<Integer> preorderTraversal(TreeNode root) {
return traversal(root,new ArrayList<Integer>());
}
public List<Integer> traversal(TreeNode node,List<Integer> list){
if (node==null) return list;
list.add(node.val);
traversal(node.left,list);
traversal(node.right,list);
return list;
}
思路:
- 由于原函数不符合递归的形式,所以创建一个子函数用于递归
- 先添加当前节点值再递归左右节点
- 设置递归终点
- 主函数调用子函数时需要传入一个ArrayList对象
迭代形式
使用栈模拟递归
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> list = new ArrayList<>();
if (root==null) return list;
Stack<TreeNode> stack = new Stack<>();
TreeNode node = root;
while (!stack.isEmpty()||node!=null){
while (node!=null){
list.add(node.val);
stack.push(node);
node = node.left;
}
node = stack.pop();
node = node.right;
}
return list;
}
思路
- 新建List对象,如果root为空,直接返回空表
- 如果root不为空,新建node指向root
- 内循环将未遍历过的左结点全部存进stack,同时完成保存数据的功能
- 弹出栈内左结点,再使node指向左结点的右结点
- 外循环执行到遍历结束
- 返回list
Morris遍历
这个估计没人想看吧,要是有人想看再call我更新。