剑指OFFER----把二叉树打印成多行

最新推荐文章于 2020-08-07 21:30:31 发布

晚风_清扬

最新推荐文章于 2020-08-07 21:30:31 发布

阅读量204

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指OFFER

本文链接：https://blog.youkuaiyun.com/RUN32875094/article/details/80148944

剑指OFFER 专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一种从上到下按层打印二叉树的方法，包括两种实现方式：循环解法和递归解法。循环解法使用队列记录每一层的节点，并通过迭代输出各层节点；递归解法则采用深度优先遍历策略，根据当前节点的深度来决定输出顺序。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
//循环解法

class Solution {
public:
        vector<vector<int> > Print(TreeNode* pRoot) {
            vector<vector<int>> v;
            if(pRoot == NULL)
                return v;
            queue<TreeNode *> qu;
            TreeNode * p = NULL;
            qu.push(pRoot);
            vector<int> t;int count = 0;
            while(!qu.empty()){
                count = qu.size();//获取每一层的节点数
                while(count--){
                    p = qu.front();
                    t.push_back(p->val);
                    if(p->left)
                        qu.push(p->left);
                    if(p->right)
                        qu.push(p->right);
                    qu.pop();
                }
                v.push_back(t);
                t.clear();
            }
            return v;
        }
};

//递归写法，深度优先遍历。利用栈的深度和树的高度对比
class Solution {
public:
        vector<vector<int> > Print(TreeNode* pRoot) {
            vector<vector<int>> v;
            if(pRoot == NULL)
                return v;
            dfs(pRoot,1,v);
            return v;
        }
        void dfs(TreeNode* root,int depth,vector<vector<int>> &v)
        {
            if(root == NULL)return ;
            if(depth > v.size())
                v.push_back(vector<int>());
            v[depth - 1].push_back(root->val);
            dfs(root->left,depth + 1, v);
            dfs(root->right,depth + 1, v);
        }
};