LeetCode Binary Tree Zigzag Level Order Traversal

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

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

0 ·

CC 4.0 BY-SA版权

文章标签：

#Leetcode #leetcode #LeetCode

LeetCode 专栏收录该内容

38 篇文章

订阅专栏

本文介绍了一种使用栈实现的二叉树锯齿形层序遍历算法，通过两个栈交替存储每层节点的方式，实现了从根节点到叶子节点的锯齿形顺序遍历。

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

/**
 * Definition for binary tree
 * 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) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        vector<vector<int> > zigzag;
        
        if (!root) {
            return zigzag;
        }
        
        stack<TreeNode *> cur, next;
        cur.push(root);
        TreeNode *temp;
        vector<int> level;
        int flag = 1;
        
        while (!cur.empty()) {
            temp = cur.top();
            cur.pop();
            level.push_back(temp->val);
                
            if (flag) {
                if (temp->left) {
                    next.push(temp->left);
                }
                    
                if (temp->right) {
                    next.push(temp->right);
                }
            }
            else {
                if (temp->right) {
                    next.push(temp->right);
                } 
                    
                if (temp->left) {
                    next.push(temp->left);
                }
            }
            
            if (cur.empty()) {
                flag = !flag;
                
                zigzag.push_back(level);
                level.clear();
                
                swap(cur, next);
            }
        }
        
        return zigzag;
    }
};