锯齿遍历

60 阅读1分钟
class Solution {
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        if (root == null) {
            return res;
        }

        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        boolean flag = true;

        while (!queue.isEmpty()) {
            int size = queue.size();
            List<Integer> levelList = new ArrayList<>();

            for (int i = 0; i < size; i++) {
                TreeNode node = queue.poll();
                if (flag) {
                    levelList.add(node.val);
                } else {
                    levelList.add(0, node.val);
                }
                if (node.left != null) {
                    queue.offer(node.left);
                }
                if (node.right != null) {
                    queue.offer(node.right);
                }
            }

            res.add(levelList);
            flag = !flag;
        }

        return res;
    }
}

在这个实现中,我们使用了一个队列 queue 来存储当前层的节点,使用了一个变量 flag 来记录当前层数的遍历顺序,使用了一个变量 levelList 来存储当前层的节点值。如果 flag 为 true,则从左往右遍历当前层,将节点值添加到 levelList 的末尾;如果 flag 为 false,则从右往左遍历当前层,将节点值添加到 levelList 的开头。每次遍历完一层后,将 flag 取反,以便下一层遍历时反转遍历顺序。希望这个实现对您有所帮助,如果您还有其他问题,请随时向我提问。