二叉树的层次遍历

来源:互联网 发布:传淘宝需要注意什么 编辑:程序博客网 时间:2024/06/03 11:21

题目:

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

样例

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

  3 / \9  20  /  \ 15   7

返回他的分层遍历结果:

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

思路:

      返回值是一个以向量为元素的向量,先在函数外定义两个向量std::vector<vector<int>> v2;和std::vector<int> v1;将一层的节点的数值放入v1,完成一个v1的放入就将v1为一个元素放入v2中,清空v1,便于下一层的节点的数进入v1中。建立一个队列暂存一层的结点,符合先进先出的原则。先将跟节点放入队列中,记录队列的节点的个数。最开始将根节点放入队列,长度为q.size(),并且循环这些次,保证了一层的节点的值进入同一个v1。将节点的数值放入v1中,如果当前节点的左节点不是空就放入队列,右节点相同,按q.soze()循环完放入v2,同时清空v1。当队列中再无节点时结束程序,反回v2。

代码:

class Solution {
    /**
     * @param root: The root of binary tree.
     * @return: Level order a list of lists of integer
     */
public:
std::vector<vector<int>> v2;
std::vector<int> v1;
queue<TreeNode*> q;
TreeNode *top;
int l;
    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;
    }
};

感想:

层次遍历用到队列是一个新的知识点,循环的方式也比较奇妙。这个题改过好几次思路,这个思路是最简洁的一个。

0 0
原创粉丝点击