Binary Tree Level Order Traversal

2014-11-24 07:35:51 · 作者: · 浏览: 0

Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).

For example:
Given binary tree {3,9,20,#,#,15,7},

    3
   / 
  9  20
    /  
   15   7

return its level order traversal as:

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

我们用深度优先去处理这个题,这样就可以很容易的利用递归,使程序显得更清晰简练,更符合人的思维模式。在DFS中,我们要记录当前的深度值,因为我们是要存储当前层的结点val信息的,所以在递归的函数中我们需要一个表示当前层的参数,除此之外我们还需要一个保存最终结果的参数的引用,当然不给引用也可以,那就把这个参数以成员变量的形式设定,如果这也不给的话,你总不能让我在类里整个全局变量吧,所以我们还是选择传参数引用的形式,因为仅仅出于一个函数的功能就给整个类家成员变量,这有点说不过去,不符合类的设计规范。当然,root参数是必备的。

这样我们的函数雏形就出来了:

 void levelNode(TreeNode *root, int level,vector
  
    >& vivec )
  
应该是这样的一种形式,那么我们接下来看,我们怎么把root代表的val值加入到vivec对应的位置呢?

首先,我们没有去计算整棵树的高度,那么我们事先无法去准确的设定vivec这个实参的确切大小,我们只能通过push_back见一层加一层。那么什么时候是新的一层了呢?我们可以通过 vivec.size() < level 来判断是否是新的一层,是新的一层,就需要push_back获得新的空间,否则就直接用下标操作添加。操作完了,就继续处理左子树和右子树。

class Solution {
public:
    void levelNode(TreeNode *root, int level,vector
  
    >& vivec )
    {
        if(!root) return;
        if(level > vivec.size())
        vivec.push_back(vector
   
    ()); vivec[level - 1].push_back(root->val); levelNode(root->left,level + 1, vivec); levelNode(root->right, level + 1, vivec); } vector
    
      > levelOrder(TreeNode *root) { vector
     
       > vivec; int level = 1; levelNode(root,level,vivec); return vivec; } };
     
    
   
  

//转载请标明出处