《剑指offer》面试题32 - III. 从上到下打印二叉树 III

问题描述

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

例如:

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

    3

   / \

  9  20

    /  \

   15   7

返回其层次遍历结果：

[

  [3],

  [20,9],

  [15,7]

]

 

提示：

节点总数 <= 1000

代码

/**

 * 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>> levelOrder(TreeNode* root) {

        vector<vector<int>> ans;

        if(!root)return ans;

        queue<TreeNode*>q;

        q.push(root);

        int flag = 0;

        while(!q.empty())

        {

            int n  = q.size();

            vector<int> tmp;

            for(int i = 0; i < n; ++i)

            {

                TreeNode* node = q.front();

                q.pop();

                tmp.push_back(node->val);

                if(node->left)q.push(node->left);

                if(node->right)q.push(node->right);

            }

            if(flag++ % 2 == 1)

                reverse(tmp.begin(),tmp.end());

            ans.push_back(tmp);

        }

        return ans;

    }

};

结果：

执行用时 :8 ms, 在所有 C++ 提交中击败了55.88%的用户

内存消耗 :12.6 MB, 在所有 C++ 提交中击败了100.00%的用户

巴特西

《剑指offer》面试题32 - III. 从上到下打印二叉树 III

问题描述

代码

最新文章

热门文章