题目链接:https://leetcode-cn.com/problems/binary-tree-preorder-traversal/
给你二叉树的根节点 root ,返回它节点值的 前序 遍历。
代码如下:
1.递归法:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* 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> ans = new ArrayList<Integer>();
dfs(ans,root);//递归调用
return ans;
}
public void dfs(List<Integer> ans,TreeNode root){
if(root == null) return;
ans.add(root.val);//根
dfs(ans,root.left);//左
dfs(ans,root.right);//右
}
}
2.迭代法:
解题思路:
它先将根节点 node 和所有的左孩子入栈并加入结果中,直至 head 为空,用一个 while 循环实现。
然后,每弹出一个栈顶元素 node,就到达它的右孩子,再将这个节点当作 node 重新按上面的步骤来一遍,直至栈为空,这里又需要一个 while 循环。
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<Integer>();
if (root == null) {
return res;
}
Stack<TreeNode> stack = new Stack<TreeNode>();
TreeNode node = root;
while (!stack.isEmpty() || node != null) {
while (node != null) {
res.add(node.val);
stack.push(node);
node = node.left;
}
node = stack.pop();
node = node.right;
}
return res;
}
}