leetcode102. 二叉树的层次遍历

给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。

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

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果:

[
  [3],
  [9,20],
  [15,7]
]

解题思路:

BFS,保证每次循环时队列里面只有这一行的节点。

class Solution {
public:
    queue<TreeNode*> Q;
    vector<vector<int>> ans;
    vector<vector<int>> levelOrder(TreeNode* root) {
        if(root==NULL)
            return ans;
        
        while(Q.empty()==false)
            Q.pop();
        Q.push(root);
          
        while(Q.empty()==false)
        {
            vector<int> vec;
            vec.clear();
            
            int len=Q.size();//统计每一行的个数
            for(int i=0;i<len;i++)
            {
                TreeNode* x=Q.front();
                vec.push_back(x->val);
                Q.pop();
    
                if(x->left)
                  Q.push(x->left);
                if(x->right)
                  Q.push(x->right);                  
           }
           ans.push_back(vec);
        }
        return ans;
    }
};

猜你喜欢

转载自blog.csdn.net/zhuixun_/article/details/80490450