在用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_ba