[每日leetcode]二叉树的锯齿形层次遍历

最新推荐文章于 2025-01-21 11:26:50 发布

summer2day

最新推荐文章于 2025-01-21 11:26:50 发布

阅读量166

点赞数

CC 4.0 BY-SA版权

分类专栏：笔试题

本文链接：https://blog.csdn.net/summer2day/article/details/97430243

笔试题专栏收录该内容

151 篇文章

订阅专栏

本文介绍了如何实现二叉树的锯齿形层次遍历，通过示例展示具体步骤，并提供相关代码实现，帮助理解该遍历方式。

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

给定一个二叉树，返回其节点值的锯齿形层次遍历。（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）。

例如：
给定二叉树 [3,9,20,null,null,15,7],

返回锯齿形层次遍历如下：

[
[3],
[20,9],
[15,7]
]

/**
 * 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>> res;
        if(root==nullptr)
            return res;
        stack<TreeNode*> st1;
        stack<TreeNode*> st2;
        st1.push(root);
        bool flag=true;
        while(1)
        {
            if(flag)
            {
                vector<int> tmp;
                while(!st1.empty())
                {
                    tmp.push_back(st1.top()->val);
                    if(st1.top()->left)
                        st2.push(st1.top()->left);
                    if(st1.top()->right)
                        st2.push(st1.top()->right);
                    st1.pop();
                    
                }
                flag=false;
                res.push_back(tmp);
                if(st2.empty())
                    break;
                
            }
            else
            {
                vector<int> tmp;
                while(!st2.empty())
                {
                    tmp.push_back(st2.top()->val);
                    if(st2.top()->right)
                        st1.push(st2.top()->right);
                    if(st2.top()->left)
                        st1.push(st2.top()->left);
                    st2.pop();
                }
                flag=true;
                res.push_back(tmp);
                if(st1.empty())
                    break;
            }
            
        }
        return res;
    }
};