剑指offer32.从上到下打印二叉树

0524

代码

class Solution {
public:
    vector<int> levelOrder(TreeNode* root) {
        vector <int>res;
        queue<TreeNode*>q;
        if(root) q.push(root);
        while(!q.empty()){
            TreeNode* node=q.front();
            res.push_back(node->val);
            q.pop();
            if(node->left!=NULL)q.push(node->left);
            if(node->right!=NULL)q.push(node->right);
        }
        return res;

    }
};

思路

层序遍历，返回一个一维数组
使用一个队列来存储每一个树节点，每次存储同一层节点值，随后立即删除。然后再去考虑左右子树的节点情况。