@Leetcode二叉树的层序遍历Ⅱ
当笔者对各种二叉树的题目有点小小自信的时候,就又被这道简单题给打败了,脑袋一片空白开始看题:
给定一个二叉树,返回其节点值自底向上的层次遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)
例如:
给定二叉树 [3,9,20,null,null,15,7],
返回其自底向上的层次遍历为:
废话少说,先来看看笔者好不容易又是看解答又是自己琢磨才瞎搞出来的代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<vector<int>> levelOrderBottom(TreeNode* root) {
vector<vector<int>> ans;
if(root==NULL)
return ans;
queue<TreeNode*> q;
q.push(root);
vector<int> floor;
int length;
while(!q.empty()){
length=q.size();
for(int i=0;i<length;i++){
if(q.front()->left)
q.push(q.front()->left);
if(q.front()->right)
q.push(q.front()->right);
floor.push_back(q.front()->val);
q.pop();
}
ans.insert(ans.begin(),floor);
floor.clear();
}
return ans;
}
};
真的感动,就这么个破题,小白笔者花了很长时间才想出来这个无比简单的思路的实现代码。但是菜归菜,总结还是要做的。我们先来看看这个困扰了笔者的“小”错误:
无论是第一张图出错还是第二张图出错,基本都是一个原因:请各位coder务必注意题干在定义数据结构的时候使用的是两层vector,而你在定义的时候只写了一层vector是肯定无法通过最后的提交的,一定要写vector<vector>,就像第二张图中指示的错误一样。当然,中间存储每一层信息的vector可以只要一层,因为不管一层还是两层存储方式都是相同的,只是提交可能出错罢了。
另外,出现第一种情况的coder请检查在出现树为空时你的返回值,这个返回必须是你定义的返回答案的那个可变长数组,而不是0或者NULL。
那么来说说这个简单得不能再简单的思路和他真正卡到你的地方。要层序遍历一棵树,最简单的方法就是从上到下一层层地找,只要这一层还有结点,就算上,听上去容易,然而本题要求倒序输出整个树层,笔者又是想用reverse又是想用栈结构存储,总想着投机取巧怎么简单怎么搞,直到笔者发现了另一个好玩的东西。
vector的insert(position,item)这个方法的功能能在可变长数组的某个位置插入一个元素,亦可以将另一个可变长数组整个插入这个位置,其后面的数据会整体后移,这就相当于起到了reverse的作用,也就使得整个题使用队列仍然可以完成任务。
解题具体思路是每一层的每个结点判断是否有子结点,由于是二叉树只要判断左右。这里要注意先判断左右并且加入再将已加入层序floor的值pop出队列,否则有可能因为队列里已无元素而提前结束循环。这样找到一个结点便加入一个结点(先左后右,注意示例),能保证一整层都被查完了之后再查下一层的所有结点,此时他们作为上一层结点的儿子结点,已经全在队列之中。
最后一个小的注意事项,每一层加入最终解的数组之后,记得clear哦~