1.
思路:
首先创建一个栈和顺序表,按照根左右的前序遍历顺序去遍历这棵树,一直往左孩子方向遍历,每遍历到一个结点就入栈并且加入到顺序表里,如果没有左孩子了,就拿出栈顶元素,看它是否有右孩子,如果有,又从右孩子开始,往左孩子方向遍历,如果没有,就继续拿出栈顶元素,如此循环,直到栈为空
class Solution {
Stack<TreeNode> stack=new Stack<>();
List<Integer> list=new ArrayList<>();
public List<Integer> preorderTraversal(TreeNode root) {
if(root==null){//如果为空树
return list;
}
TreeNode cur=root;//cur用来以前序遍历顺序来遍历树
while(cur!=null||!stack.isEmpty()){//如果此时有遍历到结点,或者栈不为空
while(cur!=null){//如果是此时有遍历到结点
stack.push(cur);
list.add(cur.val);
cur=cur.left;
}
//此时没有遍历到结点
TreeNode top=stack.pop();//拿出栈顶元素
if(top.right!=null){//如果有右孩子
cur=top.right;//遍历右孩子
}
}
return list;
}
}
2.
思路:
同上题一样,只不过由根左右变成左根右,所以往顺序表里面添加元素的地方要继续调整
class Solution {
List<Integer> list=new ArrayList<>();
Stack<TreeNode> stack=new Stack<>();
public List<Integer> inorderTraversal(TreeNode root) {
if(root==null){
return list;
}
TreeNode cur=root;
while(cur!=null||!stack.isEmpty()){
while(cur!=null){
stack.push(cur);
cur=cur.left;
}
TreeNode top=stack.pop();
list.add(top.val);//说明没有左孩子,到了左根右的根,打印
if(top.right!=null){
cur=top.right;
}
}
return list;
}
}
3.
思路:
因为后序遍历是左右根,所以先遍历到的是根,但最后输出的才是根,贴合栈这种“先进后出”的数据结构,所以要采用到栈这个数据结构
根据左右根的顺序,先一直让左孩子结点全部入栈,如果左孩子为null,则看看现在栈顶元素是否有右孩子,没有就出栈,有就将右孩子入栈,再次循环
但是这里有可能出现死循环,就是下图这种情况
当遍历到A,左孩子为null,此时A为栈顶元素,A有右孩子B,所以B入栈,然后B没有左右孩子,所以B出栈,此时A没有左孩子,A为栈顶元素,A有右孩子B,B又入栈,然后B没有左右孩子,B又出栈……
所以我们还有添加一种判断情况,即已经入栈再出栈了的我们不再入栈,所以要创建一个结点来记录入栈再出栈了的这种结点,如果是B这种结点,则栈顶元素A继续出栈而不是让B这种结点再入栈
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
//记录后序遍历的数组
List<Integer> list=new ArrayList<>();
//如果为空结点,直接返回空数组
if(root==null){
return list;
}
Stack<TreeNode> stack=new Stack<>();
TreeNode cur=root;
TreeNode prev=null;//用来记录已经入栈又出栈了的结点
//如果结点不为空,或者栈不为空
while(cur!=null||!stack.isEmpty()){
//将左孩子全部入栈或者没有左孩子但是有右孩子的
while(cur!=null){
stack.push(cur);
cur=cur.left;
}
//此时cur为null,看看栈顶元素
TreeNode top=stack.peek();
//如果栈顶元素也没有右孩子,或者右边已经入栈又出栈了的
if(top.right==null||top.right==prev){
//将栈顶元素输出并出栈
list.add(top.val);
stack.pop();
//记录当前结点为入栈又出栈了的
prev=top;
}else{
//有右孩子且不是入栈又出栈了的
cur=top.right;
}
}
return list;
}
}