力扣练习——42 二叉树的层次遍历 II

1.问题描述

给定一个二叉树,返回其节点值自底向上的层次遍历。 (即按从叶子节点所在层到根节点所在的层,逐层从左向右遍历)

 

例如:

给定二叉树 [3,9,20,null,null,15,7],

    3

   / \

  9  20

     /  \

   15   7

返回其自底向上的层次遍历为:

[

  [15,7],

  [9,20],

  [3]

]

程序输出为:

15 7 9 20 3

 

可使用以下main函数:

#include <iostream>

#include <queue>

#include <cstdlib>

#include <cstring>

#include <algorithm>

using namespace std;

struct TreeNode

{

    int val;

    TreeNode *left;

    TreeNode *right;

    TreeNode() : val(0), left(NULL), right(NULL) {}

    TreeNode(int x) : val(x), left(NULL), right(NULL) {}

    TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}

};

TreeNode* inputTree()

{

    int n,count=0;

    char item[100];

    cin>>n;

    if (n==0)

        return NULL;

    cin>>item;

    TreeNode* root = new TreeNode(atoi(item));

    count++;

    queue<TreeNode*> nodeQueue;

    nodeQueue.push(root);

    while (count<n)

    {

        TreeNode* node = nodeQueue.front();

        nodeQueue.pop();

        cin>>item;

        count++;

        if (strcmp(item,"null")!=0)

        {

            int leftNumber = atoi(item);

            node->left = new TreeNode(leftNumber);

            nodeQueue.push(node->left);

        }

        if (count==n)

            break;

        cin>>item;

        count++;

        if (strcmp(item,"null")!=0)

        {

            int rightNumber = atoi(item);

            node->right = new TreeNode(rightNumber);

            nodeQueue.push(node->right);

        }

    }

    return root;

}

 

int main()

{

    TreeNode* root;

    root=inputTree();

    vector<vector<int> > res=Solution().levelOrderBottom(root);

    for(int i=0; i<res.size(); i++)

    {

        vector<int> v=res[i];

        for(int j=0; j<v.size(); j++)

            cout<<v[j]<<" ";

    }

}

2.输入说明

首先输入结点的数目n(注意,这里的结点包括题中的null空结点)

然后输入n个结点的数据,需要填充为空的结点,输入null。

3.输出说明

输出结果,每个数据的后面跟一个空格。

4.范例

输入

7
3 9 20 null null 15 7

输出

15 7 9 20 3 

5.代码

#include <iostream>
#include <queue>
#include <cstdlib>
#include <cstring>
#include<set>
#include<algorithm>
using namespace std;



struct TreeNode

{

    int val;

    TreeNode *left;

    TreeNode *right;

    TreeNode() : val(0), left(NULL), right(NULL) {}

    TreeNode(int x) : val(x), left(NULL), right(NULL) {}

    TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}

};

TreeNode* inputTree()

{

    int n, count = 0;

    char item[100];

    cin >> n;

    if (n == 0)

        return NULL;

    cin >> item;

    TreeNode* root = new TreeNode(atoi(item));

    count++;

    queue<TreeNode*> nodeQueue;

    nodeQueue.push(root);

    while (count < n)

    {

        TreeNode* node = nodeQueue.front();

        nodeQueue.pop();

        cin >> item;

        count++;

        if (strcmp(item, "null") != 0)

        {

            int leftNumber = atoi(item);

            node->left = new TreeNode(leftNumber);

            nodeQueue.push(node->left);

        }

        if (count == n)

            break;

        cin >> item;

        count++;

        if (strcmp(item, "null") != 0)

        {

            int rightNumber = atoi(item);

            node->right = new TreeNode(rightNumber);

            nodeQueue.push(node->right);

        }

    }

    return root;

}

vector<vector<int> > levelOrderBottom(TreeNode* root)
{
    //1.定义返回数组
    vector<vector<int> > levelOrder;
    if (root == NULL) return levelOrder;//空树直接返回
    //2.层次遍历,使用队列
    queue<TreeNode*>q;
    q.push(root);//入队列
    //遍历队列
    while (!q.empty())
    {
        //定义遍历每层后存储的数组
        vector<int>level;
        int size = q.size();
        for (int i = 0; i < size; i++)
        {
            TreeNode * node = q.front();//队头元素
            q.pop();//出队
            level.push_back(node->val);//存储入动态数组level中
            if (node->left)
                q.push(node->left);//左子树入队列
            if (node->right)
                q.push(node->right);//右子树入队列

        }
        levelOrder.push_back(level);//记录每层内容
    }
    //由于上面遍历结果是正向的层次遍历,而输出要求从下往上,因此要逆转
    reverse(levelOrder.begin(), levelOrder.end());
    return levelOrder;
}


int main()

{

    TreeNode* root;

    root = inputTree();

    vector<vector<int> > res = levelOrderBottom(root);

    for (int i = 0; i < res.size(); i++)

    {

        vector<int> v = res[i];

        for (int j = 0; j < v.size(); j++)

            cout << v[j] << " ";

    }

}

 

posted @ 2022-07-27 16:31  努力奋斗的小企鹅  阅读(39)  评论(0)    收藏  举报