L2-011 玩转二叉树 (25 分)

最新推荐文章于 2024-07-26 22:53:48 发布

狼狐

最新推荐文章于 2024-07-26 22:53:48 发布

阅读量80

点赞数

分类专栏：天梯

本文链接：https://blog.csdn.net/langhuya/article/details/117047298

版权

天梯专栏收录该内容

98 篇文章

订阅专栏

本文介绍了一种使用递归方法实现二叉树层次遍历的算法，并提供了完整的 C++ 代码示例。该算法首先定义了节点的层级，然后通过前序遍历的方式递归地填充每个层级的节点值。

摘要由CSDN通过智能技术生成

原题链接

#include <bits/stdc++.h>
using namespace std;
vector<int> level(10000, -1);
vector<int> in, pre;
void leveorder(int root, int start, int end, int index)
{
    if (start > end)
        return;
    int i = start;
    while (i < end && in[i] != pre[root])
    {
        i++;
    }
    level[index] = pre[root];
    leveorder(root + 1, start, i - 1, 2 * index + 2);
    leveorder(root + 1 + i - start, i + 1, end, index * 2 + 1);
}
int main()
{
    int n, cnt = 0;
    cin >> n;
    in.resize(n);
    pre.reserve(n);
    for (int i = 0; i < n; i++)
    {
        cin >> in[i];
    }
    for (int i = 0; i < n; i++)
    {
        cin >> pre[i];
    }
    leveorder(0, 0, n - 1, 0);
    for (int i = 0; i < level.size(); i++)
    {
        if (level[i] != -1 && cnt != n - 1)
        {
            cout  << level[i]<<" ";
            cnt++;
        }
        else if (level[i] != -1)
        {
            cout << level[i];
            break;
        }
    }
    return 0;
}