题目:
给定一个二叉树,返回它的 前序 遍历。
示例:
输入: [1,null,2,3]
1
\
2
/
3
输出: [1,2,3]
进阶: 递归算法很简单,你可以通过迭代算法完成吗?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-preorder-traversal
方法一
思路:
递归算法。
代码如下:
List<Integer> res = new ArrayList<>();
/**
* 二叉树的前序遍历
* 递归实现
* 借助辅助函数
* @param root
* @return
*/
public List<Integer> preorderTraversal (TreeNode root) {
helper(root);
return res;
}
/**
* 这种时候需要借助辅助函数
* @param root
*/
public void helper (TreeNode root) {
if (root == null) return;
res.add(root.val);
preorderTraversal(root.left);
preorderTraversal(root.right);
}
我有点不懂,写在一个函数里,显示空指针错。
我觉得这样应该可以的,但是我不知道该怎么写,留一下。
/**
* 不借助辅助函数
* @param root
* @return
*/
public List<Integer> preorderTraversal2 (TreeNode root) {
List<Integer> list = new ArrayList<>();
if (root == null) ;// 这里有问题
list.add(root.val);
preorderTraversal(root.left);
preorderTraversal(root.right);
return list;
}
方法二
思路:
栈
代码如下:
/**
* 二叉树的前序遍历
* @param root
* @return
*/
public List<Integer> preorderTraversal3 (TreeNode root) {
List<Integer> res = new ArrayList<>();
if (root == null) return res;
// 栈
Stack<TreeNode> stack = new Stack<>();
stack.push(root);
while (!stack.isEmpty()) {
TreeNode cur = stack.pop();
if (cur == null) continue;
res.add(cur.val);
// 这样栈中左子树在上
stack.push(cur.right);
stack.push(cur.left);
}
return res;
}