剑指offer---从上往下打印二叉树

题目：从上往下打印二叉树

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

* 注意： 此题只是要求返回一维数组，若需要返回二维数组（即按照层序遍历的顺序，每一层的节点存放到一个一维数组，最后多层组合为二维数组），请点击这里

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/
class Solution {
public:
    vector<int> PrintFromTopToBottom(TreeNode* root) {

    }
};

解题代码：

class Solution {
public:
    vector<int> PrintFromTopToBottom(TreeNode* root) {
        vector<int> res;
        if(root == nullptr)
            return res;

        TreeNode* p = root;
        queue<TreeNode*> q;
        q.push(root);

        while(!q.empty()){
            TreeNode* temp = q.front();
            res.push_back(temp->val);
            q.pop();

            if(temp->left != nullptr)
                q.push(temp->left);
            if(temp->right)
                q.push(temp->right);
        }
        return res;
    }
};