LeetCode 103: Binary Tree Zigzag Level Order Traversal

最新推荐文章于 2021-02-05 11:10:39 发布

原创最新推荐文章于 2021-02-05 11:10:39 发布 · 237 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #二叉树 #层次遍历

leetcode 专栏收录该内容

91 篇文章

订阅专栏

本文介绍了一种二叉树的锯齿形层序遍历算法，通过该算法可以实现从根节点开始，按层级顺序输出节点值，并在每个层级交替改变输出方向。文章提供了一个C++实现示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Binary Tree Zigzag Level Order Traversal

Given a binary tree, return the zigzag level order traversal of its nodes’ values. (ie, from left to right, then right to left for the next level and alternate between).

For example:
Given binary tree {3,9,20,#,#,15,7},

return its zigzag level order traversal as:

解题思路

二叉树的层次遍历，和 Binary Tree Level Order Traversal 的不同之处主要在处理每一层时需要考虑是从左往右输出还是从右往左输出。代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
        vector<vector<int>> result;

        if (root == NULL) return result;

        queue<TreeNode *> nodeQueue;
        nodeQueue.push(root);
        bool leftToRight = true;
        while (!nodeQueue.empty()) {
            int level_size = nodeQueue.size(); // Note: 判断每层的元素个数
            vector<int> level_values(level_size); // 预留 level_size 个元素的大小
            for (int i = 0; i < level_size; ++i) {
                TreeNode *front = nodeQueue.front();
                nodeQueue.pop();

                // 从左向右 or 从右向左
                if (leftToRight) {
                    level_values[i] = front->val;
                }
                else {
                    level_values[level_size-i-1] = front->val;
                }

                if (front->left != NULL) nodeQueue.push(front->left);
                if (front->right != NULL) nodeQueue.push(front->right);
            }
            // 方向反转
            leftToRight = leftToRight ^ 1;
            result.push_back(level_values);
        }

        return result;
    }
};