**题目描述:**求给定的二叉树的前序遍历。
**分析:**与前一个题目相似,前一个题目是使用两个栈进行二叉树的后序遍历。这次根据前序与后续遍历之间的位置关系,决定采用一个队列加一个栈进行二叉树的前序遍历。
后续遍历是左子树右子树根节点(左右中),对其使用两个栈的原因是由于二叉树从上到下一次遍历最后输出的是最上层的节点,最先输出的是叶子节点,因此将上层的节点按照逆序入站最后弹出得到的就是后序遍历。
前序遍历是根节点左子树右子树(中左右),对前序遍历使用队列是因为前序是从上到下进行输出,先输出根节点最后输出叶子节点。
两个题目都使用了辅助栈,用来记录中间没有遍历子树的节点。
import java.util.*;
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public ArrayList<Integer> preorderTraversal(TreeNode root) {
ArrayList<Integer> arr=new ArrayList<Integer>();
if(root==null){
return arr;
}
Queue<TreeNode> queue = new LinkedList<TreeNode>();
Stack<TreeNode> stack2=new Stack<TreeNode>();//记录没有遍历右子树的节点
TreeNode cur=root;
while(cur!=null||!stack2.isEmpty()){
while(cur!=null){
queue.offer(cur);//从上到下一次进入队列,队列使用offer函数入列
stack2.push(cur);//记录中间没有访问全子树的节点
cur=cur.left;
}
cur=stack2.pop();
cur=cur.right;
}
while(!queue.isEmpty()){arr.add(queue.poll().val);}//队列将数据出列使用poll函数
return arr;
}
}