剑指offer：按层打印二叉树

题目描述

从上往下打印出二叉树的每个节点，同层节点从左至右打印

解题思路：

想到的是最简单的用二维数组存储每一层的数值，之后把二维数组拉平就是结果。测试通过

看到用队列的方法觉得很不错，学习。 在把该节点值加进结果数组的时候，该节点的左子树和右子树分别进入队列，从队列中取节点进行下面操作

代码：

class Solution {
public:
    vector<vector<int>> temp;
    
    void deeptree(TreeNode* root,int d){
        
        if(root==NULL)return;
       if(temp.size()==d) temp.push_back(vector<int>());//不加该句会发生错误，显示越界
        temp[d].push_back(root->val);
        
        deeptree(root->left,d+1);
        deeptree(root->right,d+1);
        
    }
    
    vector<int> PrintFromTopToBottom(TreeNode* root) {

        vector<int> res;
        
         deeptree(root,0);
        int n=temp.size();
        for(int i=0;i<n;i++)
            for(int j=0;j<temp[i].size();j++){
                
                res.push_back(temp[i][j]);
            }
        return res;
    }
};

用队列思想解决的代码

class Solution {
public:
    vector<int> PrintFromTopToBottom(TreeNode *rt) {
        queue<TreeNode*> q;
        q.push(rt);
        vector<int> r;
        while(!q.empty()){
            rt = q.front(); q.pop();
            if(!rt) continue;
            r.push_back(rt -> val);
            q.push(rt -> left);
            q.push(rt -> right);
        }
        return r;
    }
};