从上到下打印二叉树

题目描述

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

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

从上到下打印二叉树_第1张图片

返回:

[3,9,20,15,7]

算法思想

建立一个vector数组ret用来当做返回的结果数组,建立一个队列用来接收二叉树的内容,当队列不为空时,取出队列的队头元素temp,将其val数值压入结果数组中,然后再将该队列队头元素弹出,最后依次访问temp的左右子树,若不为空,压入队列中,进行下一步的判断和结果数组的压入。

代码实现

class Solution {
public:

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

你可能感兴趣的:(LeetCode,算法,leetcode)