思路:前序遍历,也就是“根,左,右”的顺序去遍历,递归思路简单,不细说了,把res,结果集合设置成全局变量就行
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ class Solution { List<Integer> res=new ArrayList<>(); public List<Integer> preorderTraversal(TreeNode root) { TreeNode cur=root; if(cur!=null) { res.add(cur.val); preorderTraversal(cur.left); preorderTraversal(cur.right); } return res; } }
思路;不递归,那就还是设置一个栈,
从根节点开始,每次迭代弹出当前栈顶元素,并将其孩子节点压入栈中,先压右孩子再压左孩子。此时要保证压入的不为空(因为栈,先入后出,要想先遍历左子树,必须要让左子节点后一个被压入栈)
在这个算法中,输出到最终结果的顺序按照 根-左,右,符合前序遍历的顺序。
class Solution { List<Integer> res=new ArrayList<>(); public List<Integer> preorderTraversal(TreeNode root) { Stack<TreeNode> stack=new Stack<>(); if(root==null)return res; stack.push(root);//每次都要先弹出栈顶元素,所以说要先在循环外把根压入 while(!stack.isEmpty()) { TreeNode cur=stack.pop(); res.add(cur.val); if(cur.right!=null) stack.push(cur.right); if(cur.left!=null) stack.push(cur.left); } return res; } }