剑指offer 从上往下打印二叉树（C++）

题目描述

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

解题思路

就是二叉树的层序遍历序列。

代码实现

/*
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) {
        if(root == NULL)
            return {};
        vector<int> v;
        queue<TreeNode*> q;
        q.push(root);
        while(!q.empty())
        {
            v.push_back(q.front()->val);
            if(q.front()->left) // 注意要对结点的左右子树是否存在进行判断
                q.push(q.front()->left); 
            if(q.front()->right)
                q.push(q.front()->right);
            q.pop();
        }
        return v;
    }
};