【Java语言】力扣系列----剑指 Offer 32 - III. 从上到下打印二叉树 III

请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。

例如:
给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
返回其层次遍历结果:

[
  [3],
  [20,9],
  [15,7]
]
 

提示:

节点总数 <= 1000

具体代码实现如下:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
       // 双栈法,利用栈的先进后出特点,(没有下面那个方法快)
       List<List<Integer>> res = new LinkedList<>();
       Stack<TreeNode> odd = new Stack<>();// 奇数栈
       Stack<TreeNode> even = new Stack<>();// 偶数栈
       if(root == null) return res;// 非空判断
       odd.push(root);// 将根节点压入奇数栈
       int count = 1;// 用于计录层数
       // 当奇数栈和偶数栈均为空时跳出
       while(!odd.isEmpty() || !even.isEmpty()){
           // 用于记录每层的结果
        ArrayList<Integer> list = new ArrayList<>();
        if(count % 2 == 1){// 奇数层
        // 奇数层非空时
            while(!odd.isEmpty()){
                TreeNode cur = odd.pop();
                list.add(cur.val);
                // 将下一层的节点加入到偶数栈中,因为倒叙,刚好符合栈的先进后出特点
                if(cur.left != null) even.push(cur.left);
                if(cur.right != null) even.push(cur.right);
            }
        }else{// 偶数层
        // 偶数层非空
            while(!even.isEmpty()){
                TreeNode cur = even.pop();
                list.add(cur.val);
                // 将下一层节点加入到奇数栈中,因为奇数层是正序,所以先加右节点
                if(cur.right != null) odd.push(cur.right);
                if(cur.left != null) odd.push(cur.left);
            }
        }
        count++;
        res.add(list);
       }
       return res;
     
        /*
        // 注意后面只需要new一个LinkeList<>();即可
        List> res = new LinkedList<>();
        // 创建队列,利用其先进先出的特点,对二叉树进行BFS遍历
        Queue queue = new LinkedList<>();
        // 判断根节点是否为空,注意,这里不能直接返回null
        if(root == null)  return res;
        // 将根节点加入到队列中
        queue.add(root);
        // 用于反转时判断
        boolean flag = true;
        // 当队列为空时,跳出循环
        while(!queue.isEmpty()){
            // 因为要分层打印,故先回去每一层的长度
            int len = queue.size();
            // 因为要返回一个list集合的类型,故创建一个小的list,用于存储每一层的结果
            List list = new ArrayList<>();
            // 通过循环可遍历打印每一层的元素
            for(int i = 0; i < len; i++){
            TreeNode cur = queue.poll();
            list.add(cur.val);
            if(cur.left != null) queue.add(cur.left);
            if(cur.right != null) queue.add(cur.right);
            }
            // 偶数行进行反转
            if(!flag) Collections.reverse(list);
            // 将结果接入到大的list集合中
            res.add(list);
            flag = !flag;
        }
        return res;
        */
    }
}

人生若只如初见,何事秋风悲画扇。
等闲变却故人心,却道故人心易变。
-----------纳兰性德

小白寄语:学如逆水行舟,不进则退。

你可能感兴趣的:(数据结构与算法基础,leetcode,剑指offer)