浙大数据结构:03-树2 List Leaves

这道题我借用了一点上一题的代码思路,这题考察的主要是层序遍历,即用队列来实现,当然此处我依然采用数组模拟队列来实现。
机翻

1、条件准备

map的键存下标,后面值分别存左右子树的下标,没有子树就存-1.
head数组只是为了找到树的跟结点在哪,实现原理后面再说
q数组是模拟队列,front是队头下标,rear是队尾下标
cpp 复制代码
#include <iostream>
#include <map>
using namespace std;

map<int, pair<int, int>> m;
int head[15];
int q[50], front = 0, rear = -1;
主函数先是加快输入输出,h存根节点的下标,traval是进行层序遍历并输出
cpp 复制代码
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);
    int h;
    h = init(m);
    traval(h);

    return 0;
}

2、init函数

先输入结点数n,然后循环输入数据,每组数据的子结点存到map里面,如果没有子节点存-1.
然后标记子节点出现过的数字,用head数组存,最后遍历head数组,没有标记的就是根节点。
因为根节点不会出现在任一组数据中,所以可以这样找到。
cpp 复制代码
int init(map<int, pair<int, int>> &m)
{
    int n;
    cin >> n;
    for (int i = 0; i < n; i++)
    {
        char f, s;
        cin >> f >> s;
        m[i].first = f == '-' ? -1 : f - '0';
        m[i].second = s == '-' ? -1 : s - '0';
        if (m[i].first != -1)
            head[m[i].first] = 1;
        if (m[i].second != -1)
            head[m[i].second] = 1;
    }
    for (int i = 0; i < n; i++)
    {
        if (head[i] == 0)
            return i;
    }
    return -1;
}

3、traval函数

用队列先把根节点放入,flag是为了调整输出格式的。
遍历队列,取出队头,如果为叶子结点则输出,只有第一次输出不加空格。
然后若该结点有子节点,放入队列,然后弹出队头。
cpp 复制代码
void  traval(int h)
{
    q[++rear] = h;
    int flag = 1;
    while (front <= rear)
    {
        int t = q[front];
        if (m[t].first == -1 && m[t].second == -1)
        {
            if (flag)
               { cout <<t;flag=0;}
        else
        cout << ' ' << t;
        }
        if (m[t].first != -1)
            q[++rear] = m[t].first;
        if (m[t].second != -1)
            q[++rear] = m[t].second;
        front++;
    }
}

4、总结

这道题相对而言没有那么难了,还是较为友好的。
完整代码如下:
cpp 复制代码
#include <iostream>
#include <map>
using namespace std;

map<int, pair<int, int>> m;
int head[15];
int q[50], front = 0, rear = -1;

int init(map<int, pair<int, int>> &m)
{
    int n;
    cin >> n;
    for (int i = 0; i < n; i++)
    {
        char f, s;
        cin >> f >> s;
        m[i].first = f == '-' ? -1 : f - '0';
        m[i].second = s == '-' ? -1 : s - '0';
        if (m[i].first != -1)
            head[m[i].first] = 1;
        if (m[i].second != -1)
            head[m[i].second] = 1;
    }
    for (int i = 0; i < n; i++)
    {
        if (head[i] == 0)
            return i;
    }
    return -1;
}

void  traval(int h)
{
    q[++rear] = h;
    int flag = 1;
    while (front <= rear)
    {
        int t = q[front];
        if (m[t].first == -1 && m[t].second == -1)
        {
            if (flag)
               { cout <<t;flag=0;}
        else
        cout << ' ' << t;
        }
        if (m[t].first != -1)
            q[++rear] = m[t].first;
        if (m[t].second != -1)
            q[++rear] = m[t].second;
        front++;
    }
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);
    int h;
    h = init(m);
    traval(h);

    return 0;
}
相关推荐
汉克老师1 小时前
GESP2026年9月认证C++七级( 第三部分编程题(2、括号序列))精讲
c++·gesp·小学生·学c++编程
w_zero_one1 小时前
链表(4)
java·数据结构·算法
白山编程大哥2 小时前
C语言篇:语法进阶
c语言·数据结构·算法
jimy13 小时前
虚函数vtable,链接时看到构造函数,linker把符号地址写入vtable(三)
开发语言·c++
Logic1013 小时前
C语言/数据结构动态规划题解:爬楼梯(含坏台阶)——避开障碍物的路径计数
c语言·数据结构·动态规划·时间复杂度·空间复杂度·算法题·爬楼梯
潼心1412o4 小时前
C++初阶(长期更新)第9讲:string(上)
开发语言·c++
朝朝辞暮i4 小时前
C++ 第 39 章: 阶段性总复习——类、对象、构造、继承、this、指针与智能指针
开发语言·c++·算法·ros2
wuminyu4 小时前
Virtual Thread重投递至ForkJoinPool任务队列过程解析
java·linux·c语言·jvm·c++
All for pursuit.4 小时前
【回溯-6】79.单词搜索
数据结构·c++·算法·leetcode
Benny_Tang5 小时前
AT_arc180_d [ARC180D] Division into 3 题解
数据结构·c++·算法