main.cpp - main - silenww/laba

main.cpp
92 строки2 KB
silenww
initial commit
10 мар 2026, 16:43
10 мар 2026, 16:433cae890
#include <iostream>
#include <clocale>
#include "btree.h"
#include "windows.h"
using namespace std;

void menu() {
    cout << "\nЛР5. Вариант 2\n";
    cout << "1) Добавить запись\n";
    cout << "2) Найти запись\n";
    cout << "3) Удалить запись\n";
    cout << "4) Средний оклад для стажа 12 лет\n";
    cout << "5) Печать дерева в ширину\n";
    cout << "6) Загрузить из файла\n";
    cout << "7) Тесты\n";
    cout << "0) Выход\n";
    cout << "Выбор: ";
}

int main() {
    SetConsoleCP(1251);
    SetConsoleOutputCP(1251);    

    BtreeNode* root = nullptr;

    while (true) {
        menu();

        int cmd;
        cin >> cmd;

        if (!cin) {
            clearInput();
            cout << "Ошибка ввода.\n";
            continue;
        }

        if (cmd == 0)
            break;

        if (cmd == 1) {
            Data d = inputData();
            root = addNode(root, d);
            cout << "Запись добавлена.\n";
        }
        else if (cmd == 2) {
            int id;
            cout << "Введите ID: ";
            cin >> id;

            BtreeNode* p = findNode(root, id);

            if (p == nullptr)
                cout << "Не найдено.\n";
            else
                printData(p->myData);
        }
        else if (cmd == 3) {
            int id;
            cout << "Введите ID: ";
            cin >> id;

            if (findNode(root, id) == nullptr)
                cout << "Не найдено.\n";
            else {
                root = deleteNode(root, id);
                cout << "Удалено.\n";
            }
        }
        else if (cmd == 4) {
            averageSalaryFor12(root);
        }
        else if (cmd == 5) {
            BFS(root);
        }
        else if (cmd == 6) {
            string filename;
            cout << "Введите имя файла: ";
            cin >> filename;
            loadFromFile(filename, root);
        }
        else if (cmd == 7) {
            runTests();
        }
        else {
            cout << "Неизвестная команда.\n";
        }
    }
    clearTree(root);
    cout << "Выход.\n";
    return 0;
}