LintCode 69 二叉树的层次遍历

题目:solveNQueenss

要求:

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

样例:

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

3
/ \
9  20
/  \
15   7
返回他的分层遍历结果：

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

算法要求:

挑战1：只使用一个队列去实现它

挑战2：用DFS算法来做

解题思路:

用i来记录当前层遍历了几个，n来记录当前层有几个，j来记录一共遍历了几个有效结点，size表示结点数。

用DFS的话，二个参数一个是当前结点，一个是当前层数。

算法如下:

int getSize(TreeNode *root) {
if (root == NULL) {
return 1;
}
return getSize(root->left) + getSize(root->right);
}
vector<vector<int> > levelOrder(TreeNode *root) {
// write your code here
vector<vector<int> > vec;
if (root == NULL) {
return vec;
}
int size = getSize(root) - 1;
queue<TreeNode*> que;
que.push(root);
vector<int> *v = new vector<int>();
int n = 1;
int i = 0;
int j = 0;
while (!que.empty() && j < size) {
TreeNode *now = que.front();
que.pop();
i++;
if (now) {
v->push_back(now->val);
j++;
}
if (i >= n) {
i = 0;
n *= 2;
if (!v->empty()) {
vec.push_back(*v);
v = new vector<int>();
}
}
if (now) {
que.push(now->left);
que.push(now->right);
} else{
que.push(NULL);
que.push(NULL);
}
}
if (!v->empty()) {
vec.push_back(*v);
}
return vec;
}