剑指offer22--从上到下打印二叉树

题目描述

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

思路

即二叉树的层序遍历。
用一个队列来辅助实现从上到下，从左到右的顺序——先将根结点入栈，当栈不空的时候，将队首结点的值打印输出，遍历左右子树结点入队，进行下一轮。直到最后队列空了。

实现

class Solution {
public:
    vector<int> PrintFromTopToBottom(TreeNode* root) {
        if (!root) return {};
        
        vector<int> v;
        queue<TreeNode*> q;//辅助队列
        q.push(root);//先将根结点入队
        while (!q.empty())
        {
            v.push_back(q.front()->val);//打印对手结点的值
            if (q.front()->left)//左子树入队
                q.push(q.front()->left);
            if (q.front()->right)//右子树入队
                q.push(q.front()->right);
            q.pop();//弹出队首的值准备进行下一轮循环
        }
        return v;
    }
};

拓展

再来看看两个进阶版的层序遍历：
进阶版：按层返回
把二叉树打印成多行

进阶版plus：按之字形打印
之字形打印二叉树