剑指32-1 从上到下打印二叉树

从上到下打印出二叉树的每个节点，同一层的节点按照从左到右的顺序打印。

 

例如:
给定二叉树: [3,9,20,null,null,15,7],

3
/ \
9 20
/ \
15 7
返回：

[3,9,20,15,7]

 

因为是要从上到下，所以要求先进先出，先压入的一层就要先输出，所以采用队列。

从第一层开始，先输出根节点的值，然后将左右子树压入队列（如果不为空）。

之后每一层，先记录下来队列的长度，即为这一层节点的数量，然后挨个弹出，对该层每个节点都输出值，然后压入左右子树（如果不为空）。因为已经提前记录了这一层节点数量，不会输出到下一层去。

注意处理根节点为空的情况。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<int> levelOrder(TreeNode* root) {
        if(root==nullptr)
            return {};
        queue<TreeNode*> q;
        vector<int> ret;
        ret.push_back(root->val);
        if(root->left!=nullptr)
            q.push(root->left);
        if(root->right!=nullptr)
            q.push(root->right);
        while(!q.empty()){
            int cur_size=q.size();
            while(cur_size!=0){
                TreeNode* curptr=q.front();
                q.pop();
                cur_size--;
                ret.push_back(curptr->val);
                if(curptr->left!=nullptr)
                    q.push(curptr->left);
                if(curptr->right!=nullptr)
                    q.push(curptr->right);
            }
        }
        return ret;
    }
};

 经过思考发现，记录每一层数量其实完全是白用功，因为本题不用分层输出，直接顺序到下一层完全没影响。