剑指offer | 训练题58：按之字形打印二叉树

最新推荐文章于 2020-04-23 20:05:41 发布

原创最新推荐文章于 2020-04-23 20:05:41 发布 · 218 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

69 篇文章

订阅专栏

本文介绍了一种使用BFS方法实现的之字形打印二叉树算法。该算法通过新建队列来循环处理每一层节点，并根据层数的奇偶性决定是否翻转输出顺序。

题目描述

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

思路

/**
 * 思路：BFS，新建一个队列，将每一层放进队列里循环读出，按照单数层和双数层决定要不要翻转vector
 */
/*
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>> ans;
        if (pRoot == nullptr)
            return ans;
        queue<TreeNode*> q;
        int level = 0;
        q.push(pRoot);
        while (!q.empty()) {
            int size = q.size();
            vector<int> levelelem;
            level++;
            while (size--) {
                TreeNode* t = q.front();
                q.pop();
                levelelem.push_back(t->val);
                if (t->left != nullptr)
                    q.push(t->left);
                if (t->right!= nullptr)
                    q.push(t->right);
            }
             if (level%2!=0)
                    ans.push_back(levelelem);
                else {
                    reverse(levelelem.begin(),levelelem.end());
                    ans.push_back(levelelem);
                }
         }
        return ans;
    }

};