从上往下打印二叉树

剑指offer面试题

题目描述
从上往下打印出二叉树的每个节点,同层节点从左至右打印。

分析
1、题目是考察二叉树的遍历
2、按层次遍历
3、从根结点开始打印
3、把根节点的左右子结点存储下来,才能接着遍历左右子结点的子节点。
4、还要依次把根节点的左右子结点出来遍历其子节点
5、由此,我们可以想到要用队列来存储结点,满足先入先出

由此我们找到规律:
1、每一次打印一个结点的时候,如果该结点有子结点,则把该结点的子结点放到一个队列的末尾。
2、然后,到队列的头部取出最早进入队列的结点,重复前面的打印操作,知道队列中所欲的结点都被打印出来为止

/*
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) {
        vector<int> result;     //定义一个数组保存结点的值
        queue q;     //定义一个队列保存结点

        if(root == NULL)
            return result;    

        q.push(root);

        //队列不空时
        while(!q.empty())
        {
            result.push_back(q.front()->val);     //队头压入数组

            if(q.front()->left != NULL)          //左子入队
                q.push(q.front()->left);

            if(q.front()->right != NULL)         //左子入队
                q.push(q.front()->right);

            q.pop();
        }
        return result;
    }
};

你可能感兴趣的:(数据结构和算法)