在用vector实现二叉树层次遍历时遇到的问题
在用c++编写二叉树层次遍历的算法时,作为一个连入门都算不上的菜鸟,遇到了很多问题,绞尽脑汁后终于算是蹩脚地解决了,谨以此篇博客记录分享下(望各位大牛多指教批评~在此抱拳了!!!)
我的二叉树结点结构如下:
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
我代码的层次遍历函数声明如下:
vector<vector<int>> levelOrder(TreeNode* root){}
返回值为一个二维向量,这也就为后面我遇到麻烦埋下伏笔,所幸都解决了。之所以要用二维向量作返回值是为了能够按层输出元素,在此说下我的函数中主要的容器(用来实现队列从而达到层次遍历目的)及作用
vector<vector> sort; //二维向量,存储不同层上的元素值。
vector<TreeNode *> que; //存储结点
vectordepth; //存储结点深度,便于分层
下面说下遇到的第一个问题:在如何将二叉树里的元素根据层数存入时遇到了困难,想了很久,最后想出用一个专门存储结点深度的容器vectordepth 在将每个二叉树结点存入队列que时,将其深度也存入depth队列;从这以后只要que队列有元素出队,depth队列也同时出队,入队操作也是如此,这样就自然而然的将不同元素分层了;下面附上解决了这个问题后的问题代码:
vector<vector<int>> levelOrder(TreeNode* root) {
if (root != NULL) {
int t = 0;int i; //记录队列中弹出结点的深度,即层数
que.push_back(root);
depth.push_back(t);
while (!que.empty()) {
k = que.front();
i = depth.front();
t = i + 1;
que.erase(que.begin());
depth.erase(depth.begin());
sort[i].push_back(k->val);
if (k->left != NULL) { que.push_back(k->left); depth.push_back(t); }
if (k->right != NULL) { que.push_back(k->right); depth.push_back(t); }
}
}
return sort;
}
当时很天真的以为这样就完事儿了,结果报错,存在数组访问越位巴拉巴拉巴拉之类的错误;后来发现,在使用sort[i]时,忽略了分配内存的问题,后来想了想,最终用push_back()函数来解决了。
下面放上最终版本代码:
class Solution {
vector<vector<int>> sort; //二维向量,存储不同层上的元素值
vector<TreeNode *> que; //存储结点
vector<int>depth; //存储结点深度,便于分层
TreeNode *k;
public:
vector<vector<int>> levelOrder(TreeNode* root) {
if (root != NULL) {
int t = 0;int i; //记录队列中弹出结点的深度,即层数
que.push_back(root);
depth.push_back(t);
sort.push_back(vector<int>()); //!!!!!!!!!!!!分配内存
while (!que.empty()) {
k = que.front();
i = depth.front(); i
f(sort.size()!=(i+1)) sort.push_back(vector<int>()); //若二维向量行大小不等于当前层数,则再申请一行!!!
t = i + 1;
que.erase(que.begin());
depth.erase(depth.begin());
sort[i].push_back(k->val);
if (k->left != NULL) { que.push_back(k->left); depth.push_back(t); }
if (k->right != NULL) { que.push_back(k->right); depth.push_back(t); } }
}
return sort; }
};