您的位置:首页 > Web前端

剑指offer:按之字形顺序打印二叉树

2018-03-22 10:39 351 查看
题目:请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
分析:读完题,可以确定本题是用树的层次遍历做,无非就是变形的层次遍历。
首先,第一行从左到右顺序打印,即根结点。从第二层开始,需要从右往左顺序打印,即先打印右结点,在打印左结点,看起来是层次遍历的先进后出,因此可以用栈来实现。打印第三层时,需要从左往右打印,则最右结点应先进栈,最左结点后进栈,因此可以确保左结点先出栈,从左往右打印。值得注意,应该用两个栈来分别保存从左往右打印的和从右往左打印的。
当打印的层数为奇数层时,是从左往右打印,则它们的子节点(位于偶数层)应该从左往右进栈,因为它们是从右往左打印;
当打印的层数为偶数层时,是从右往左打印,则它们的子节点(位于奇数层)应该从右往左进栈,因为它们是从左往右打印。package Solution47;

import java.util.ArrayList;
import java.util.LinkedList;

//请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
public class Solution {
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;

public TreeNode(int val) {
this.val = val;
}
}

public ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> Print=new ArrayList<>();
if(pRoot==null) return Print;
LinkedList<TreeNode> linkedList1=new LinkedList<>(); //保存从左往右进栈的子节点
LinkedList<TreeNode> linkedList2=new LinkedList<>(); //保存从右往左进栈的子节点
linkedList2.push(pRoot);
int count=0,nextcount=1,depth=1;
ArrayList<Integer> subprint=new ArrayList<>();
while(linkedList1.size()!=0 || linkedList2.size()!=0){
if(depth%2==1) {//打印奇数层,下一层节点从左往右进栈
TreeNode top=linkedList2.pop();
subprint.add(top.val);
count++;
if(top.left!=null){
linkedList1.push(top.left);
}
if(top.right!=null){
linkedList1.push(top.right);
}
if(count==nextcount){
depth++;
count=0;
nextcount=linkedList1.size();
Print.add(subprint);
subprint=new ArrayList<>();
}
}else{ //打印偶数层,下一层节点从右往左进栈
TreeNode top=linkedList1.pop();
subprint.add(top.val);
count++;
if(top.right!=null){
linkedList2.push(top.right);
}
if(top.left!=null){
linkedList2.push(top.left);
}
if(count==nextcount){
depth++;
count=0;
nextcount=linkedList2.size();
Print.add(subprint);
subprint=new ArrayList<>();
}
}
}
return Print;
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: