lintcode二叉树的层次遍历

来源:互联网 发布:东方挂馅炒饭 知乎 编辑:程序博客网 时间:2024/06/02 05:57

1.描述:

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

样例

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

  3 / \9  20  /  \ 15   7

返回他的分层遍历结果:

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

2.思路:先判断根节点不为空时将根节点放入队列q中,while  q不为空时,令l等于q的长度,whilel不等于0,将q的头结点赋给top,然后把top的值加入vector v1中,然后判断根节点的左右孩子是否为空,不为空就入队列,然后使队列尾节点出队,当l循环到等于0时,当前层节点值就全部加入v1中,并将下一层节点全部入队,然后在把v1加到v2中,并清空v1,当循环到q为空时,整个树就遍历结束

3.代码:

vector<vector<int>> levelOrder(TreeNode *root) {
        vector<vector<int>> v2;
        vector<int> v1;
        queue<TreeNode*> q;
        TreeNode *top;
        int l;
        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;
 
    }  

};

4.感想:

递归用多了一次不用还有点想不出思路,借助队列的方式来层次遍历二叉树感觉这思路真的很新奇

0 0
原创粉丝点击