牛客網——劍指offer 從上往下打印二叉樹

題目描述
從上往下打印出二叉樹的每個節點，同層節點從左至右打印。

vector<int> PrintFromTopToBottom(TreeNode* root) {
       queue<TreeNode*> que;
       vector<int> arr;
       if(root == NULL) return arr;
        
        que.push(root);
        while(!que.empty())
        {
            TreeNode* flag;
            flag = que.front();
            que.pop();
            arr.push_back(flag->val);
            if(flag->left != NULL)que.push(flag->left);
            if(flag->right != NULL)que.push(flag->right);
        }
        
        return arr;
    }