二叉树的层次遍历

时间:2023-01-26 17:27:10
题目:

给出一棵二叉树,返回其节点值的层次遍历(逐层从左往右访问)

样例

给一棵二叉树 {3,9,20,#,#,15,7} :

  3
 / \
9  20
  /  \
 15   7

返回他的分层遍历结果:

[
  [3],
  [9,20],
  [15,7]
]
代码:
class Solution {
    std::vector<vector<int>> v2;
    std::vector<int> v1;
    queue<TreeNode*> q;
    TreeNode *top;
    int l;
    /**
     * @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
        if(root==NULL)
       return v2;
       q.push(root);
       while(!q.empty())
       {
          l=q.size();
          while(l!=0)
          {
              top=q.front();
              v1.push_back(top->val);
              l--;
              if(top->left!=NULL)
              q.push(top->left);
              if(top->right!=NULL)
              q.push(top->right);
              q.pop();
          }
          v2.push_back(v1);
          v1.clear();
       }
       return v2;
    }
};
感想:这题实在借鉴了别人的代码后写出的,思路是返回值是一个以向量为元素的向量,先在函数外定义两个向量std::vector<vector<int>> v2;和std::vector<int> v1;将一层的节点的数值放入v1,完成一个v1的放入就将v1为一个元素放入v2中,清空v1,便于下一层的节点的数进入v1中。建立一个队列暂存一层的结点,符合先进先出的原则。先将跟节点放入队列中,记录队列的节点的个数。最开始将根节点放入队列,长度为q.size(),并且循环这些次,保证了一层的节点的值进入同一个v1。将节点的数值放入v1中,如果当前节点的左节点不是空就放入队列,右节点相同,按q.soze()循环完放入v2,同时清空v1。当队列中再无节点时结束程序,反回v2。