牛客题霸NC14二叉树的之字形层序遍历Java题解

牛客题霸NC14二叉树的之字形层序遍历Java题解

https://www.nowcoder.com/practice/47e1687126fa461e8a3aff8632aa5559?tpId=117&&tqId=34935&rp=1&ru=/ta/job-code-high&qru=/ta/job-code-high/question-ranking

方法：利用队列
解题思路：将节点加入到队列中，利用队列Queue的先进后出，依此弹出节点。将每次弹出节点的值保存到一个ArrayList中（tmp）,如果当前层是奇数层（从1开始），执行尾插，如果当前层是偶数层，执行头插。如果弹出的节点有左、右子节点，则将左、右子节点加入到队列Queue中，每一层遍历完，将这一层的节点都加入到ArrayList<ArrayList<integer>> list 中。

import java.util.*;
 
/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 * }
 */
 
public class Solution {
    /**
     *
     * @param root TreeNode类
     * @return int整型ArrayList<ArrayList<>>
     */
    public ArrayList<ArrayList<Integer>> zigzagLevelOrder (TreeNode root) {
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
 
        if(root!=null){
            queue.add(root);
        }
 
        while(!queue.isEmpty()){
            ArrayList<Integer> tmp = new ArrayList<>();     //存储每一层节点
            for(int i= queue.size();i>0;i--){               //遍历当前层的节点
                TreeNode node = queue.poll();               //弹出队列中的节点
 
                if((res.size()+1)%2!=0){     //res.size()+1：当前的层数，从1开始
                    tmp.add(node.val);       // 奇数层 -> 尾插
                }else{
                    tmp.add(0,node.val);     // 偶数层 -> 头插
                }
 
                if(node.left!=null){         //如果左子节点不为空，则将其加入到队列中
                    queue.add(node.left);
                }
                if(node.right!=null){         //如果左子节点不为空，则将其加入到队列中
                    queue.add(node.right);
                }
            }
            res.add(tmp);               //将这一层的节点加入到res中
        }
        return res;
    }
}