给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
例如:
给定二叉树: [3,9,20,null,null,15,7],
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
/*
思路:
1、 定义一个二维容器,每行存一层:vector <vector > ret;
2、 首先第一层(根)入队,并记录队列的大小,
对队内的每个元素都进行左右子节点放入队尾的操作,
其中一个存完子节点之后即可出队,放入ret_temp,
即队列存入了下一层。
3、 ret_temp再放入ret,由此就得到了一层
4、 对下一次的执行1~3的操作
*/
/**
* 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>> ret;
queue<TreeNode*> buf;
int depth=-1;
if(root==NULL)
return ret;
buf.push(root);
while(!buf.empty()){
int count=buf.size();
vector <int> ret_temp;
while(count>0){
TreeNode* cur=buf.front();
ret_temp.push_back(cur->val);
buf.pop();
if(cur->left!=NULL)
buf.push(cur->left);
if(cur->right!=NULL)
buf.push(cur->right);
count--;
}
ret.push_back(ret_temp);
}
return ret;
}
};