我的思路是,将同一层的树的节点都保存下来,这样当然可以,时间复杂度会小一点,但是空间复杂度会大很多;
当时没有想到,用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;
}
};