[数据结构与算法] LeetCode 103 二叉树的锯齿形层序遍历

开发: C++知识库 Java知识库 JavaScript Python PHP知识库人工智能区块链大数据移动开发嵌入式开发工具数据结构与算法开发测试游戏开发网络协议系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑笔记本显卡显示器固态硬盘硬盘耳机手机 iphone vivo oppo 小米华为单反装机图拉丁

-> 数据结构与算法 -> LeetCode 103 二叉树的锯齿形层序遍历 -> 正文阅读

[数据结构与算法]LeetCode 103 二叉树的锯齿形层序遍历

定义一个变量isFromLeft

如果isFromLeft为true，则将节点的值添加到列表的尾部

否则，添加到列表的头部?

if(isFromLeft) {
    oneLevel.add(cur.val);
} else {
    oneLevel.add(0 , cur.val);
}

遍历完当前层之后，移动到下一层，则从相反的方向开始访问当前层的所有节点

isFromLeft = !isFromLeft;

完整代码

class Solution {
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        LinkedList<List<Integer>> result = new LinkedList<>();
        if(root == null) {
            return result;
        }
        LinkedList<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        boolean isFromLeft = true;
        while(!queue.isEmpty()) {
            List<Integer> oneLevel = new ArrayList<>();
            int size = queue.size();
            
            for(int i = 0; i < size; i++) {
                TreeNode cur = queue.poll();
                if(isFromLeft) {
                    oneLevel.add(cur.val);
                } else {
                    oneLevel.add(0 , cur.val);
                }
                if(cur.left != null) {
                    queue.add(cur.left);
                }
                if(cur.right != null) {
                    queue.add(cur.right);
                }
            }
            isFromLeft = !isFromLeft;
            result.add(oneLevel);
        }
        return result;
    }
}