LintCode-二叉树的层次遍历

来源:互联网 发布:根据ip地址查域名 编辑:程序博客网 时间:2024/04/28 21:31

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

样例:

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

  3 / \9  20  /  \ 15   7

返回他的分层遍历结果:

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

做题思路:创建队列,把根节点放入队列,建立一个length用来记录队列长度,把树的结点指针指向队列中的第一个数,取值放入数组1中,把队列的数弹出,然后长度随之减少,根节点的左孩子如果不是空,就放入队列,右孩子不为空则放入队列,放入数组后弹出,清空数组,再继续循环,数组1放入数组2中,直到队列为空。

关键代码:

class Solution {
    /**
     * @param root: The root of binary tree.
     * @return: Level order a list of lists of integer
     */
public:
    std::vector<int> f;
    std::vector<vector<int>> w;
    vector<vector<int>> levelOrder(TreeNode *root) {
        // write your code here
    if(root==NULL) return w;
    queue<TreeNode *> q;
    q.push(root);
    int length;
    while(!q.empty())
     { length=q.size();
       while(length!=0)
        {  TreeNode *t=q.front();
          f.push_back(t->val);
          q.pop();
          length--;
          if(t->left!=NULL)
           q.push(t->left);
           if(t->right!=NULL)
           q.push(t->right);
        }
      w.push_back(f);
      f.clear();
     }
     return w;
    }
};

做题感想:刚开始自己不会写,后来上网查了查,然后自己按照理解重新写了,前几遍输出的答案和期望答案不一样,后来给同学讲了一遍,讲的过程中突然发现,小vector一直没有清空,加了clear的语句之后就对了。

0 0
原创粉丝点击