## 在用vector实现二叉树层次遍历时遇到的问题

在用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;	}
     };
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值