剑指 Offer 32 - III. 从上到下打印二叉树 III

请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。

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

        3
          /  \
                          9   20
                               /  \
                            15   7
返回其层次遍历结果:[[3],[20,9],[15,7]]

=====================================================

相比32-II加了一个判断,判断是插在数组的前面还是后面

复制代码
class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> v;
        if (root == NULL)
            return v;
        queue<TreeNode *> q;
        q.push(root);
        int i = 0;
        while (!q.empty()) {
            q.push(NULL);
            TreeNode* r = q.front();
            vector<int> t;
            while (r != NULL) {
                if (i % 2 == 0)
                    t.push_back(r->val);
                else
                    t.insert(t.begin(), r->val);
                if (r->left)
                    q.push(r->left);
                if (r->right)
                    q.push(r->right);
                q.pop();
                r = q.front();
            }
            q.pop();
            if (t.size() != 0) {
                v.push_back(t);
                i++;
            }
        }
        return v;
    }
};
复制代码

看题解,这个题目貌似是想考察双端队列,但是没有什么必要性,其实是双端队列自己不怎么熟悉,又熟悉了一下,c++的deque不太好做,看到一个大神写的用双栈实现的类似双端丢列的写法,跪了

复制代码
class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        if(root==NULL)      return {};
        stack<TreeNode*> s1; //管理第0层等偶数层
        stack<TreeNode*> s2; // 管理第1层等奇数层
        vector<vector<int>> res;
        s1.push(root);
        int count = 0;
        while(!s1.empty() || !s2.empty()){
            vector<int> temp;
            if(count%2==0){
                int len = s1.size();
                for(int i=0;i<len;i++){
                    temp.push_back(s1.top()->val);
                    if(s1.top()->left)    s2.push(s1.top()->left);
                    if(s1.top()->right)     s2.push(s1.top()->right);
                    s1.pop();
                }
            }
            else{
                int len = s2.size();
                for(int i=0;i<len;i++){
                    temp.push_back(s2.top()->val);
                    if(s2.top()->right)    s1.push(s2.top()->right);
                    if(s2.top()->left)     s1.push(s2.top()->left);
                    s2.pop();
                }
            }
            count++;
            res.push_back(temp);
        }
        return res;
    }
};
复制代码

 

posted on   4小旧  阅读(20)  评论(0编辑  收藏  举报

相关博文:
阅读排行:
· 终于写完轮子一部分:tcp代理 了,记录一下
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
< 2025年3月 >
23 24 25 26 27 28 1
2 3 4 5 6 7 8
9 10 11 12 13 14 15
16 17 18 19 20 21 22
23 24 25 26 27 28 29
30 31 1 2 3 4 5

导航

统计

点击右上角即可分享
微信分享提示