剑指offer:从上往下打印二叉树

题目描述

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

 

AC C++ Solution:

二叉树的层次遍历,借助一个队列就可以了

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