剑指offer60 层序打印二叉树

来源:互联网 发布:朱婷 大冠军杯 知乎 编辑:程序博客网 时间:2024/06/05 18:00


从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。


/*

struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
//先进先出用队列
class Solution {
public:
        vector<vector<int> > Print(TreeNode* pRoot) {
         vector<vector<int> >vvi;//定义二维数组
            if(pRoot==NULL)
               return vvi;
             TreeNode*pNode;
            std::queue<TreeNode*>q;
              q.push(pRoot);
            while(!q.empty())
                {
                int size=q.size();
                vector<int>vi;
                while(size--)//一开始战中的数据减完,上层的数据,--size和size--的区别,--
                    //size--是先执行下面的操作再--,如果是--size就是错的
                {
                  pNode=q.front();//删除队列里面最前端的数,队列是先进先出
                  q.pop();
                  vi.push_back(pNode->val);//把队列节点放到数组中
                  if(pNode->left ) q.push(pNode->left);//把下一层的数据放进去
                  if(pNode->right) q.push(pNode->right);
              }
            if(!vi.empty())
           vvi.push_back(vi);//把保存某层数据的数组放到二维数组中
            }
    return vvi;
}
};
原创粉丝点击