LeetCode[Tree]: Binary Tree Zigzag Level Order Traversal

最新推荐文章于 2020-01-13 22:20:44 发布

chfe910

最新推荐文章于 2020-01-13 22:20:44 发布

阅读量630

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode LeetCode 文章标签：递归 LeetCode 迭代 ZigZag 算法

本文链接：https://blog.youkuaiyun.com/chfe007/article/details/42172633

LeetCode 同时被 2 个专栏收录

74 篇文章

订阅专栏

LeetCode

72 篇文章

订阅专栏

本文介绍了使用递归和迭代方法实现二叉树的Zigzag层次遍历，详细解释了算法逻辑并提供了代码实现。通过比较两种方法的时间性能，展示了其实现细节和效率。

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

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},
3
/ \
9 20
/ \
15 7
return its zigzag level order traversal as:
[
[3],
[20,9],
[15,7]
]

我的迭代解法如下：

vector<vector<int> > zigzagLevelOrder(TreeNode *root) {
    vector<vector<int> > res;

    vector<TreeNode *> iter;
    if (root) iter.push_back(root);

    int isForward = 1;
    while (!iter.empty()) {
        int size = iter.size();
        vector<int> level;

        for (int i = 0; i < size; ++i) {
            level.push_back(iter[i]->val);
            if (iter[i]->left) iter.push_back(iter[i]->left);
            if (iter[i]->right) iter.push_back(iter[i]->right);
        }

        if (!isForward) reverse(level.begin(), level.end());
        res.push_back(level);
        iter.erase(iter.begin(), iter.begin() + size);
        isForward ^= 0x01;
    }

    return res;
}

我的递归解法如下：

class Solution {
public:
    vector<vector<int> > zigzagLevelOrder(TreeNode *root) {
        res.clear();
        if (!root) return res;
        zigzagLevelOrder(root, 1);
        return res;
    }

private:
    vector<vector<int> > res;

    void zigzagLevelOrder(TreeNode *root, int level) {
        if (res.size() < level) res.resize(level);
        if (level % 2) res[level - 1].push_back(root->val);
        else res[level - 1].insert(res[level - 1].begin(), root->val);

        if (root->left) zigzagLevelOrder(root->left, level + 1);
        if (root->right) zigzagLevelOrder(root->right, level + 1);
    }
};

两种算法的时间性能相同，如下图所示：