(BFS 中等)Leetcode 102. 二叉树的层次遍历

在这里插入图片描述

我的思路是,将同一层的树的节点都保存下来,这样当然可以,时间复杂度会小一点,但是空间复杂度会大很多;

当时没有想到,用que的size当做当前的截断判断依据;

/**
 * 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) {
        queue<TreeNode*>que;
        
        vector<vector<int>>result;
        if(root==nullptr)
            return result;
        que.push(root);
        while(!que.empty())
        {
            vector<int>vec;
            vector<TreeNode*>vec_que;
            while(!que.empty())
            {
                root=que.front();
                vec.push_back(root->val);
                vec_que.push_back(root);
                que.pop();
            }
            result.push_back(vec);
            for(int i=0;i<vec_que.size();i++)
            {
                if(vec_que[i]->left)
                    que.push(vec_que[i]->left);
                if(vec_que[i]->right)
                    que.push(vec_que[i]->right);
            }
        }
        return result;
    }
  
};
发布了741 篇原创文章 · 获赞 185 · 访问量 30万+

猜你喜欢

转载自blog.csdn.net/Fire_to_cheat_/article/details/103518078