剑指Offer——从上往下打印二叉树

题目描述

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

思路：借助队列，若左孩子不空，入左孩子；右孩子不空，入右孩子。

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