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

按之字形顺序打印二叉树

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

双端队列就相当于双栈:

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
  public:
    vector<vector<int> > Print(TreeNode* pRoot) {
        vector<vector<int> > ret;
        if (pRoot == nullptr) return ret;

        deque<TreeNode*> s;
        s.push_back(pRoot);
        int level = 0;
        while (!s.empty()) {
            int size = s.size();
            vector<int> tmp;
            while (size--) {
                if (level % 2 == 0) {
                    TreeNode* p = s.back();
                    s.pop_back();
                    tmp.push_back(p->val);
                    if (p->left) s.push_front(p->left);
                    if (p->right) s.push_front(p->right);
                } else {
                    TreeNode* p = s.front();
                    s.pop_front();
                    tmp.push_back(p->val);
                    if(p->right) s.push_back(p->right);
                    if(p->left) s.push_back(p->left);
                }
            }
            ret.push_back(tmp);
            level++;
        }
        return ret;
    }

};

也可以改变每层打印顺序,用vector的insert()方法

全部评论

相关推荐

04-02 14:40
浙江大学 设计
无语😓&nbsp;就喜欢找我茬,研究生怎么了&nbsp;研究生就是天才吗&nbsp;就得所有报告文件都会,最烦做表
我推的MK:是这样的,那些领导就是自己什么都不懂就把所有东西扔给你,指望白嫖你的劳动力,如果你的表现不如预期就启动攻击学历模式,这都学不会是怎么考上浙大的
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务