【剑指offer】13.从上往下打印二叉树

总目录：

算法之旅导航目录

 

1.问题描述

 不分行从上往下打印出二叉树的每个节点，同层节点从左至右打印。例如输入{8,6,10,#,#,2,1}，如以下图中的示例二叉树，则依次打印8,6,10,2,1(空节点不打印，跳过)，请你将打印的结果存放到一个数组里面，返回。

 

 

 

2.问题分析

 1辅助队列法，参考前面两题的辅助队列做法

2递归，建立一个二维的vector，每个元素用来存储每层的节点值，递归函数需要输入当前层号

3.代码实例

辅助队列法

/*
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> printVec;

        //空树
        if (!root) {
            return printVec;
        }

        //初始化队列
        queue<TreeNode*> q;
        q.push(root);

        //初始化当前层节点计数器
        int nodeCount = 0;

        //逐层迭代
        while (!q.empty()) {
            //遍历当前层的所有节点
            nodeCount = q.size();
            for (int i = 0; i < nodeCount; i++) {
                auto* pNode = q.front();
                q.pop();

                //本层逻辑,交换当前节点的左右节点
                printVec.push_back(pNode->val);

                if (pNode->left) q.push(pNode->left);
                if (pNode->right) q.push(pNode->right);
            }

            //本层结束
        }

        return printVec;
    }
};

 递归

class Solution {
public:
    void traverse(TreeNode* root, vector<vector<int>>& res, int depth) {
        if(root){
            //新的一层
            if(res.size() < depth)  
                res.push_back(vector<int>{}); 
            //vector从0开始计数因此减1，在节点当前层的vector中插入节点
            res[depth - 1].push_back(root->val);
        }
        else
            return;
        //递归左右时进入下一层
        traverse(root->left, res, depth + 1); 
        traverse(root->right, res, depth + 1);
    }
    
    vector<int> PrintFromTopToBottom(TreeNode* root) {
        vector<int> res;
        vector<vector<int> > temp;
        if(root == NULL)
            //如果是空，则直接返回空vector
            return res; 
        traverse(root, temp, 1);
        //送入一维数组
        for(int i = 0; i < temp.size(); i++)
            for(int j = 0; j < temp[i].size(); j++)
                res.push_back(temp[i][j]);
        return res;
    }
};