Advertisement
Korotkodul

tree7_sum_subtree

Apr 19th, 2025 (edited)
226
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 4.33 KB | None | 0 0
  1. #include <iostream>
  2. #include <vector>
  3. using namespace std;
  4.  
  5. const int inf = 2e9;
  6.  
  7. struct node {
  8.     node* parent;
  9.     node* l;
  10.     node* r;
  11.     int val;
  12.  
  13.     node(node* par) : parent(par), l(nullptr), r(nullptr), val(inf) {}
  14.     node() : parent(nullptr), l(nullptr), r(nullptr), val(inf) {}
  15. };
  16.  
  17. struct tree {
  18.     //vector<int> arr; // используется ТОЛЬКО для инициализации дерева
  19.     node* main_root; // Изменен на указатель для удобства
  20.     int cur_size = 0;
  21.  
  22.     // Создает поддерево
  23.     node* create_subtree(node* par, int Li, int Ri, const vector <int> &arr) {
  24.         if (Li > Ri) return nullptr; // Проверка на правильные границы
  25.  
  26.         node* root = new node(par);
  27.         root->val = arr[(Ri + Li) / 2]; // устанавливаем значение корня
  28.  
  29.         // Создание левого и правого поддеревьев
  30.         root->l = create_subtree(root, Li, (Ri + Li) / 2 - 1, arr);
  31.         root->r = create_subtree(root, (Ri + Li) / 2 + 1, Ri, arr);
  32.  
  33.         return root; // Возвращаем указатель на созданный узел
  34.     }
  35.  
  36.     // Инициализация дерева
  37.     void create(const vector <int> &arr) {
  38.         cur_size = arr.size();
  39.         if (arr.size() == 0) {
  40.             main_root = nullptr; // В случае пустого массива
  41.         } else {
  42.             main_root = create_subtree(nullptr, 0, arr.size() - 1, arr);
  43.         }
  44.         cout << "Tree created\n";
  45.     }
  46.  
  47.     // Обход дерева (in-order)
  48.     void trav(node* root) {
  49.         if (root == nullptr) return; // Проверка на nullptr
  50.         trav(root->l); // Обход левого поддерева
  51.         cout << root->val << " "; // Обработка текущего узла
  52.         trav(root->r); // Обход правого поддерева
  53.     }
  54.  
  55.     void replace_el(node* root, int Li, int Ri, int i, int x) {
  56.         if (root == nullptr) return;
  57.  
  58.         int mid = (Li + Ri) / 2;
  59.         if (i == mid) {  // Нашли узел, который нужно заменить
  60.             root->val = x;
  61.             return;
  62.         }
  63.  
  64.         if (i < mid) {
  65.             replace_el(root->l, Li, mid - 1, i, x);  // Ищем в левом поддереве
  66.         } else {
  67.             replace_el(root->r, mid + 1, Ri, i, x);  // Ищем в правом поддереве
  68.         }
  69.     }
  70.  
  71.     void del(node* &root) {
  72.         if (root == nullptr) {
  73.             return;
  74.         }
  75.         del(root->l);
  76.         del(root->r);
  77.         if (root->l == nullptr && root->r == nullptr) {
  78.             cout << "clidren already nullptr\n";
  79.         }
  80.         cout << "del:" << root->val << "\n";
  81.         delete root;
  82.         root = nullptr;
  83.         if (root == nullptr) {
  84.             cout << "successfully deleted\n";
  85.             cout << "\n";
  86.         }
  87.     }
  88.  
  89.     int sum_subtree(node* &root) {
  90.         if (root == nullptr) {
  91.             return 0;
  92.         }
  93.         if (root->l == nullptr && root->r == nullptr) {
  94.             int res = root->val;
  95.             root->val = 0;
  96.             return res;
  97.         }
  98.         int L = sum_subtree(root->l);
  99.         int R = sum_subtree(root->r);
  100.         int old_val = root->val;
  101.         root->val = L + R;
  102.         if (root == main_root) {
  103.             root->val += old_val;
  104.         }
  105.         return  old_val + L + R;
  106.     }
  107. };
  108.  
  109. bool cmp(node* root1, node* root2) {
  110.      if  (root1 == nullptr) {
  111.         if (root2 != nullptr) {
  112.             return 0;
  113.         }
  114.         return 1;
  115.      }  else if(root2 == nullptr){
  116.          if (root1 != nullptr) {
  117.             return 0;
  118.          }
  119.          return 1;
  120.      }
  121.      if (root1->val != root2->val) {
  122.         return 0;
  123.      }
  124.      bool L = cmp(root1->l, root2->l);
  125.      bool R = cmp(root1->r, root2->r);
  126.      return L & R;
  127. }
  128.  
  129.  
  130. bool cmp_tree(tree T1, tree T2) {
  131.     bool ans = cmp(T1.main_root, T2.main_root);
  132.     return ans;
  133. }
  134.  
  135. int main() {
  136.     /*cin.tie(0);
  137.     cout.tie(0);
  138.     ios::sync_with_stdio(false);*/
  139.     tree T1, T2;
  140.     vector<int> a = {1, 2, 3, 4, 5, 6, 7, 8};
  141.     T1.create(a);
  142.     T1.trav(T1.main_root); cout << "\n";
  143.     int tree_sum = T1.sum_subtree(T1.main_root);
  144.     T1.trav(T1.main_root);
  145.     return 0;
  146. }
  147.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement