数据结构-剑指offer-把二叉树打印成多行

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

因为要按层打印,首先想到的是层次遍历。在二叉树的深度这道题中,首先应用到了层次遍历。每一层的节点值存入一个小vector c,再把小vector c存到大vector vec中,打印vec。(题目没有要求换行,所以不需要考虑换行符)

class Solution {
public:
        vector<vector<int> > Print(TreeNode* pRoot) {
            vector<vector<int>> vec;
            if(pRoot == NULL) 
                return vec;
            queue<TreeNode*> que;
            que.push(pRoot);
            while(!que.empty()){
                int depth = 0;
                int len = que.size();
                vector<int> c;
                while(depth<len){
                    depth++;
                    TreeNode* node = que.front();
                    que.pop();
                    c.push_back(node->val);
                    if(node->left)
                        que.push(node->left);
                    if(node->right)
                        que.push(node->right); 
                }
                vec.push_back(c);
            }
            return vec;
        }
};

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章