144. 二叉树的前序遍历

Posted 2021-02-20 lzh1043060917

 

 思路：前序遍历，也就是“根，左，右”的顺序去遍历，递归思路简单，不细说了，把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;
       
    }
}