http://blog.csdn.net/shiquxinkong/article/details/17758471
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<vector<int> >& vivec )
应该是这样的一种形式,那么我们接下来看,我们怎么把root代表的val值加入到vivec对应的位置呢?
首先,我们没有去计算整棵树的高度,那么我们事先无法去准确的设定vivec这个实参的确切大小,我们只能通过push_back见一层加一层。那么什么时候是新的一层了呢?我们可以通过 vivec.size() < level 来判断是否是新的一层,是新的一层,就需要push_back获得新的空间,否则就直接用下标操作添加。操作完了,就继续处理左子树和右子树。
class Solution {
public:
void levelNode(TreeNode *root, int level,vector<vector<int> >& vivec )
{
if(!root) return;
if(level > vivec.size())
vivec.push_back(vector<int>());
vivec[level - 1].push_back(root->val);
levelNode(root->left,level + 1, vivec);
levelNode(root->right, level + 1, vivec);
}
vector<vector<int> > levelOrder(TreeNode *root) {
vector<vector<int> > vivec;
int level = 1;
levelNode(root,level,vivec);
return vivec;
}
};
//转载请标明出处