Нерекурсивно подсчитать количество элементов на n уровне бинарного дерева

@CondorHCK · Регистрация: 20.11.2022

Студворк — интернет-сервис помощи студентам

Здравствуйте, нужно НЕРЕКУРСИВНО подсчитать количество элементов на n уровне бинарного дерева.
Нерекурсивный обход (с использованием стека) у меня уже есть (фция foo, которая обходит дерево сначала слева, потом справа), ее лишь нужно адаптировать под задачу.

Рекурсивный подсчет у меня есть (фция count_on_level).
Помогите, пожалуйста, буду очень благодарен.

Дерево:
8
1 2 3 4 5 6 7 8

Source.cpp

C++
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
#include <iostream>
#include <fstream>
#include <string>
//#include <stack>
#include "Tree.h"
#include "stack.h"
 
int count_on_level(Tree p, int level, int i = 0)
{
    int count = 0;
 
    if (i == level)
        count += 1;
 
    if (p->left)
        count += count_on_level(p->left, level, i + 1);
    if (p->right)
        count += count_on_level(p->right, level, i + 1);
 
    return count;
}
 
int foo(Tree t, int level, int i = 0)
{
    int count = 0;
    stack s;
    s.push(NULL);
    Tree tt = t;
    bool flag = false;
    do {
        if (tt && i != level)
        {
            s.push(tt);
            std::cout << tt->info << ' ';
            tt = tt->left;
        }
        else
        {
            if (s.Top() == NULL)
                flag = true;
            else
            {
                tt = s.Top();
                s.pop();
                tt = tt->right;
            }
        }
    } while (!flag);
 
    return count;
}
 
int main()
{
    std::ifstream file("text.txt");
    int count;
    file >> count;
    Tree root = Build_Balance(file, count);
    Print(root, 0);
    std::cout << count_on_level(root, 2);
    std::cout << '\n';
    std::cout << foo(root, 2);
    Clear(root);
    std::cin.get();
    return 0;
}

Tree.h

C++
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
#pragma once
#include <iostream>
#include <fstream>
 
using TInfo = int;
 
 
struct NODE
{
    TInfo info;
    NODE* left, * right;
    int count = 0;
    NODE() {}
    NODE(TInfo info, NODE* left = nullptr, NODE* right = nullptr) :info(info), left(left), right(right) {}
    ~NODE()
    {
        left = nullptr;
        right = nullptr;
    }
};
 
using Tree = NODE*;
 
Tree Build_Balance(std::ifstream& file, int count)
{
    Tree result = nullptr;
    if (count)
    {
        int count_left = count / 2;
        int count_right = count - count_left - 1;
        int x;
        file >> x;
        result = new NODE(x);
        result->left = Build_Balance(file, count_left);
        result->right = Build_Balance(file, count_right);
    }
    return result;
}
 
void Clear(Tree& t)
{
    if (t)
    {
        Clear(t->left);
        Clear(t->right);
        delete t;
        t = nullptr;
    }
}
 
void Print(Tree t, int level)
{
    if (t)
    {
        Print(t->right, level + 1);
        for (int i = 1; i <= level; ++i)
            std::cout << "   ";
        std::cout << t->info << '\n';
        Print(t->left, level + 1);
    }
}

@lemegeton · 29.11.2022, 00:37

zayats80888, а что оффтопик? Нормально же и по теме.

Спасибо. Действительно, нужно виртуальнее делать.

C++
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
#include <iostream>
 
template<typename T>
class Queue {
public:
    Queue() : base() {}
 
    Queue(const Queue<T> &) = delete;
 
    Queue<T> &operator=(const Queue<T> &) = delete;
 
    ~Queue() {
        while (base.next != &base) {
            delete base.next;
        }
    }
 
    void push(const T &value) {
        new ValueNode(base.prev, value);
    }
 
    const T &front() const {
        return dynamic_cast<ValueNode *>(base.next)->value;
    }
 
    void pop() {
        delete base.next;
    }
 
    bool empty() const {
        return base.next == &base;
    }
 
private:
    struct Node {
        Node() : prev(this), next(this) {}
 
        Node(Node *after) : prev(after), next(after->next) {
            prev->next = next->prev = this;
        }
 
        virtual ~Node() {
            prev->next = next;
            next->prev = prev;
        }
 
        Node *prev;
        Node *next;
    };
 
    struct ValueNode : public Node {
        T value;
 
        ValueNode(Node *after, const T &value) : Node(after), value(value) {}
    };
 
    Node base;
};
 
template<typename First, typename Second>
struct Pair {
    First first;
    Second second;
};
 
template<typename T>
struct Tree {
    Tree(const T &value) : value(value), left(nullptr), right(nullptr) {}
 
    T value;
    Tree<T> *left;
    Tree<T> *right;
};
 
template<typename T>
Tree<T> *insert(Tree<T> *root, const T &value) {
    if (root == nullptr) {
        return new Tree<T>(value);
    }
    Tree<T> *position = root;
    while (position != nullptr) {
        Tree<T> *&direction = (value < position->value) ? position->left : position->right;
        if (direction != nullptr) {
            position = direction;
        } else {
            direction = new Tree<T>(value);
            return root;
        }
    }
    return root;
}
 
template<typename T, typename Consumer>
void traverse(Tree<T> *root, Consumer consume) {
    if (root == nullptr) {
        return;
    }
    Queue<Pair<std::size_t, Tree<T> *>> queue;
    queue.push({0, root});
    do {
        Pair<std::size_t, Tree<T> *> current = queue.front();
        queue.pop();
        if (current.second->left != nullptr) queue.push({current.first + 1, current.second->left});
        if (current.second->right != nullptr) queue.push({current.first + 1, current.second->right});
        consume(current.first, current.second->value);
    } while (!queue.empty());
}
 
template<typename T>
void clear(Tree<T> *root) {
    if (root == nullptr) {
        return;
    }
    Queue<Tree<T> *> queue;
    queue.push(root);
    do {
        Tree<T> *current = queue.front();
        queue.pop();
        if (current->left != nullptr) queue.push(current->left);
        if (current->right != nullptr) queue.push(current->right);
        delete current;
    } while (!queue.empty());
}
 
int main() {
    Tree<int> *root = nullptr;
 
    root = insert(root, 5);
    root = insert(root, 3);
    root = insert(root, 8);
    root = insert(root, 2);
    root = insert(root, 4);
    root = insert(root, 7);
    root = insert(root, 9);
 
    traverse(root, [](std::size_t level, int value) {
        std::cout << level << ": " << value << std::endl;
    });
 
    std::size_t expectedLevel = 2;
    std::size_t count = 0;
    traverse(root, [&](std::size_t level, int value) {
        if (level == expectedLevel) {
            ++count;
        }
    });
 
    std::cout << "At level " << expectedLevel << " found " << count << " values.";
 
    clear(root);
    return 0;
}

Новые блоги и статьи Все статьи Все блоги /
Беседа с ИИ о программистах, недопускающих к созданию и правке кода генеративные ИИ и причины этого zorxor 21.09.2026 Раньше я радовался или получал некоторые эмоции, пусть небольшие, но всё же, от самого процесса написания кода, рекомпиляции и запуска, видя постепенное развитие программы и прочее. А теперь лень. . .	Мобильное приложение ColorStep pavlinmavlin 17.09.2026 Реализовал приложение Красный, Зеленый, Синий в Unity3d + c#. Название изменил на ColorStep. Приложение прошло модерацию и теперь доступно для скачивания. Делал его сам, шаг за шагом — и вот,. . .	Запрет дублирования строк в табличной части Maks 13.09.2026 Реализация из решения ниже выполнена на нетиповом справочнике "Нормы ТО" с табличной часть "Виды ТО", разработанного в КА2, со следующими реквизитами: - ВидТО (СправочникСсылка. ВидыТО); - ВидГСМ. . .	Скрипты Tampermonkey для CyberForum, ChatGPT, Claude и пр. Jin X 06.09.2026 Скрипты Tampermonkey для CyberForum, ChatGPT, Claude и пр. Работая с форумом и нейросетями в браузере часто хочется что-то подкорректировать или добавить какого-то функционала. Ниже прикреплён. . .
Программа опроса у.з. расходомера SLS-720F Argus19 02.09.2026 Программа опроса у. з. расходомера SLS-720F Программа опрашивает один раз в минуту три ультразвуковых расходомера SLS-720F через интерфейс RS-485 по протоколу Modbus RTU. Опрашиваются регистры. . .	Hyper-V: Компьютер должен поддерживать доверенный платформенный модуль 2.0. Maks 31.08.2026 При установке Windows 11 на виртуальную машину Hyper-V 2-го поколения вылезла такая ошибка: Решение: в параметрах виртуальной машины, в разделе "Безопасность" (Security) активировать флаг. . .	Архитектура биовида Стива в Майнкрафте: Зачем бонобо кубический каннибализм anaschu 30.08.2026 Кубический Вагинокапитализм в Minecraft: Математический инвариант ОДУ и рок Стивов-бонобо Главная задача разработанной «Модели Всего» — наглядно продемонстрировать наличие системной «судьбы». . .	Оттачиваю умение писать js программы. russiannick 30.08.2026 Проектом выходного дня стало написание Книги шифров Виженера. Итогом стала версия 200, синий туман. Синий туман назван так, потому что замораживает текст под собой. Нажатие синих кнопок управляют. . .