问题描述:
对一棵二叉树进行层次遍历
实现思路:
利用队列,在队尾增加,队头删除。在访问每一层时,先记住这一层的rear ,然后访问该层的节点,同时访问结点的左右子树增加在在队尾。当访问到该层的最后一个节点时,记住该队列rear的值。进行下一层的访问。
代码:
/**
* Definition of TreeNode:
* class TreeNode {
* public:
* int val;
* TreeNode *left, *right;
* TreeNode(int val) {
* this->val = val;
* this->left = this->right = NULL;
* }
* }
*/
class Solution {
/**
* @param root: The root of binary tree.
* @return: Level order a list of lists of integer
*/
public:
vector<vector<int>> levelOrder(TreeNode *root) {
// write your code here
TreeNode *Q[100000];
TreeNode *q;
vector<vector<int>>vv;
int front,rear;
int a;
front=rear=-1;
if(root==NULL) return vv;
Q[++rear]=root;
while(front!=rear) {
vector<int> v;
while(front<a) {
q=Q[++front];
v.push_back(q->val);
if(q->left!=NULL) Q[++rear]=q->left;
if(q->right!=NULL) Q[++rear]=q->right;}
vv.push_back(v);
a=rear;
}
return vv;
}
};
感想:
进行层次遍历时,利用队列,要先记住该层的节点访问完时此时a=rear的值,同时访问该层节点时还要把该层节点的左右儿子的节点的值加入队列中同时rear变化。当访问完该层节点时,同时a=此时的rear的值。接着访问下一层。