题解 | #按之字形顺序打印二叉树#

按之字形顺序打印二叉树

https://www.nowcoder.com/practice/91b69814117f4e8097390d107d2efbe0

/**
 * struct TreeNode {
 *  int val;
 *  struct TreeNode *left;
 *  struct TreeNode *right;
 *  TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
#include <algorithm>
#include <queue>
#include <vector>
class Solution {
  public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param pRoot TreeNode类
     * @return int整型vector<vector<>>
     */
    vector<vector<int> > Print(TreeNode* pRoot) {
        // write code here
        vector<vector<int>> ans;
        if (pRoot == NULL) {
            return ans;
        }
        queue<TreeNode*> q;
        q.push(pRoot);
        int i = 1;
        bool reverse_flag = false;
        while (!q.empty()) {
            //记录下这一行的值
            vector<int> row;
            int n = q.size();

            //把这一行的数值放入row中
            for (int j = 0; j < n; j++) {
                TreeNode* cur = q.front();
                q.pop();
                row.push_back(cur->val);
                if (cur->left) {
                    q.push(cur->left);
                }
                if (cur->right) {
                    q.push(cur->right);
                }
            }
            //当前行按照对应的规则放完,更新参数
            if (reverse_flag) {
                reverse(row.begin(), row.end());
            }
            ans.push_back(row);
            reverse_flag = !reverse_flag;
        }
        return ans;
    }
};

之字形遍历二叉树

思路:

按照之前一样借助队列实现层序遍历

再每一层需要反转的时候,将row vector通过函数reverse(rwo.begin(), row.end)实现反转,然后push back到最大vector中

最终循环完毕,返回两层vector答案

全部评论

相关推荐

不愿透露姓名的神秘牛友
07-09 11:15
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
07-07 13:35
虽然不怎么光彩,经过这件事,可能我真的要去认同“面试八股文早该淘汰!不会用AI作弊的程序员=新时代文盲!”这句话了
HellowordX:Ai的出现是解放劳动力的,不是用来破坏公平竞争环境的,这样下去,轻则取消所有线上面试,严重了会影响整个行业对所有人产生影响,企业会拉高入职考核各种离谱考核会层出不穷
你找工作的时候用AI吗?
点赞 评论 收藏
分享
06-07 19:59
门头沟学院 C++
补药卡我啊😭:都快15年前的了还在11新特性
你的简历改到第几版了
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务