Бинарные деревья. Составить программу, которая содержит информацию о книгах в библиотеке. - C++ - Ответ 13042017

#include <string>
#include <iostream>
#include <conio.h>
#include <cstdlib>
#include <windows.h>
 
using namespace std;
struct Rozm {
    int ID;
    string PIB, book_name, year_w, count_book;
};
// узол дерева
struct Node {
    Rozm *d;
    Node *left;
    Node *right;
};
// Прототипы функции
Node * first(Rozm *d); // Формирование первого элемента дерева
Node * search_Insert(Node *root, Rozm *d); // Поиск с добавлением
Node * add(Node *root);//Добавление нового элемента дерева
void print_tree(Node *root, int l); //Удаление дерева с удалением
void print_search(Node *root, Rozm *d, int f, string n); //Вывод дерева за критериями
void Delete_Node_BinaryTree(Node **p, int f); //Удаление вершин
 
//----------------
//MAIN
int main(int argc, char** argv) {
    SetConsoleOutputCP(1251);
    do{
        system("CLS"); 
        cout << endl << "List of library books:\n" << endl;
    int ID[10] = { 00, 15, 21, 17, 46, 51, 18, 13, 89, 75};
    string PIB[10] = { "Remark","Pushkin","King","Sapkovsky","Strugatckiy","Koelyo", "Akunin","Bronte","Frankbaum","Levi" };
    string book_name[10] = { "Triumf Ark","Balada","Dream Catcher","Witcher","To Moon and Back","Alone", "Jane Eir","FairyTailes","3 days in the forest","Big Foot" };
    string year_w[10] = { "1987","1879","1995","2001","1987","1878", "1983","1992","1874","1965" };
    string count_book[10] = { "66","25","95","20","98","18", "3","19","14","65" };
    
    // Формирование узла корня
    Rozm *d = new Rozm;
     d->ID = ID[0]; d->PIB = PIB[0]; d->book_name = book_name[0]; d->year_w = year_w[0];d->count_book = count_book[0];
    Node *root = first(d);
 
    // Формирование дерева упорядоченного по коду книги
    for (int i = 1; i < 10; i++)
    {
            Rozm *d = new Rozm;
        d->ID = ID[i]; // Случайное число от 400 до 1000
        d->PIB = PIB[i];
        d->book_name = book_name[i];
        d->year_w = year_w[i];
        d->year_w = year_w[i];
        d->count_book = count_book[i];
    
        // если такой записи не существует,
        search_Insert(root, d); // присоеденение к дереву
 
    }
    print_tree(root, 0); // Вывод дерева на экран 
    cout << endl;
    
    //ADD NEW
    add(root); // Добавление нового элемента
    cout << endl;
    print_tree(root, 0); // Вывод дерева на экран 
    cout << endl;
    
    //DELETE BOOK
    cout << "-----------------------------------------------------------------------" << endl;
    cout << "Input ID of book you want to delete: ";
    int key1; // Разговор с таким кодом будет удален
    cin >> key1;
    Delete_Node_BinaryTree(&root, key1);
    cout << endl;
    cout << endl << "New list o books: " << endl;
    print_tree(root, 0);
    cout << "-----------------------------------------------------------------------" << endl;
    cout<< "Do you want to continue:\n 1. Yes. \n 2. No." << endl;
    }while(getche()!='2');
    _getch();
    return 0;
}
 
// Формирование первого элемента дерева
Node * first(Rozm *d)
{
    Node *pv = new Node;
    pv->d = d;
    pv->left = NULL;
    pv->right = NULL;
    return pv;
}
 
// Добавление нового элемента
Node * add(Node *root)
{
    string nid, npib, nname, nyear, ncount;
    int nf;
    Rozm *d = new Rozm;
    cout << "-----------------------------------------------------------------------" << endl;
    cout << "Input data about new book: " << endl;
        cout << "ID book: " << endl;
    cin >> nf; d->ID = nf;
    cin.get();
        cout << "Name autor: " << endl;
    cin >> npib; 
    d->PIB = npib;
        cout << "Name book: " << endl;
    cin.get();
    getline(cin, nname); 
    d->book_name = nname;
        cout << "Year: " << endl;
    cin.get();
    getline(cin, nyear); 
    d->year_w = nyear;
        cout << "Count of books: " << endl;
    cin.get();
    getline(cin, ncount); 
    d->year_w = ncount;
    search_Insert(root, d);
 
}
// Поиск 
Node * search_Insert(Node *root, Rozm *d)
{
    Node *pv = root, *prev;
    bool found = false;
    while (pv && !found)
    {
        prev = pv;
        if (d->PIB == pv->d->PIB)
            found = true;
        else
            if (d->ID < prev->d->ID)
                pv = pv->left;
            else
                pv = pv->right;
 
    }
    if (found) return pv;
    // Создание нового узла
    Node *pnew = new Node;
    pnew->d = d;
    pnew->left = 0;
    pnew->right = 0;
    if (d->ID < prev->d->ID)
        // Присоединение к левому поддереву предка       
prev->left = pnew;
    else
        // Присоединение к правому поддереву предка
        prev->right = pnew;
    return NULL;
}
//Прямой обход и вывод дерева
void print_tree(Node *p, int level)
{
    if (p) {
        print_tree(p->left, level + 1); // вывод левого поддерева
        for (int i = 0; i < level; i++)
            cout << " ";
        cout << p->d->ID << " " << p->d->PIB << " " << p->d->book_name << " " << p->d->year_w << " " << p->d->count_book << endl; // Вывод кореня поддерева
        print_tree(p->right, level + 1); // вывод правого поддерева
    }
}
//Прямой обход и вывод дерева по критериям 
void print_search(Node *root, Rozm *d, int f, string n)
{
    Node *pv = root, *prev;
    bool found = false;
    while (pv && !found)
    {
        prev = pv;
        if ((n.compare(pv->d->year_w) == 0) && (f == pv->d->ID))
            found = true;
        else
            if (d->ID < prev->d->ID)
                pv = pv->left;
            else
                pv = pv->right;
    }
    if (found) cout << pv->d->ID << " " << pv->d->PIB << " " << pv->d->book_name << " " << pv->d->year_w << " " << pv->d->count_book << endl;
}
//Вывод вершин бинарногодерева
void Delete_Node_BinaryTree(Node **p, int f)
{
    if ((*p) != NULL)
    {
        if (((*p)->d->ID == f)) {
            Node * ptr = *p;
            if ((*p)->left == NULL && (*p)->right == NULL)
                *p = NULL;
            else
                if ((*p)->left == NULL)
                    *p = ptr->right;
                else
                    if ((*p)->right == NULL)
                        *p = ptr->left;
                    else
                    {
                        *p = ptr->right;
                        Node **ptr1;
                        ptr1 = p;
                        while (*ptr1 != NULL)
                            ptr1 = &((*ptr1)->left);
                        *ptr1 = ptr->left;
                    }
            delete(ptr);
 
            Delete_Node_BinaryTree(p, f);
        }
        else {
            Delete_Node_BinaryTree(&((*p)->left), f);
            Delete_Node_BinaryTree(&((*p)->right), f);
        }
    }
}