mirror of
https://github.com/krahets/hello-algo.git
synced 2026-07-17 17:26:06 +00:00
build
This commit is contained in:
@@ -460,28 +460,7 @@ AVL-дерево одновременно является и двоичным
|
||||
=== "Ruby"
|
||||
|
||||
```ruby title="avl_tree.rb"
|
||||
=begin
|
||||
File: avl_tree.rb
|
||||
Created Time: 2024-04-17
|
||||
Author: Xuan Khoa Tu Nguyen (ngxktuzkai2000@gmail.com)
|
||||
=end
|
||||
|
||||
require_relative '../utils/tree_node'
|
||||
require_relative '../utils/print_util'
|
||||
|
||||
# ## AVL-дерево ###
|
||||
class AVLTree
|
||||
# ## Конструктор ###
|
||||
def initialize
|
||||
@root = nil
|
||||
end
|
||||
|
||||
# ## Получение корневого узла двоичного дерева ###
|
||||
def get_root
|
||||
@root
|
||||
end
|
||||
|
||||
# ## Получить высоту узла ###
|
||||
### Получить высоту узла ###
|
||||
def height(node)
|
||||
# Высота пустого узла равна -1, высота листового узла равна 0
|
||||
return node.height unless node.nil?
|
||||
@@ -489,36 +468,7 @@ AVL-дерево одновременно является и двоичным
|
||||
-1
|
||||
end
|
||||
|
||||
=begin
|
||||
File: avl_tree.rb
|
||||
Created Time: 2024-04-17
|
||||
Author: Xuan Khoa Tu Nguyen (ngxktuzkai2000@gmail.com)
|
||||
=end
|
||||
|
||||
require_relative '../utils/tree_node'
|
||||
require_relative '../utils/print_util'
|
||||
|
||||
# ## AVL-дерево ###
|
||||
class AVLTree
|
||||
# ## Конструктор ###
|
||||
def initialize
|
||||
@root = nil
|
||||
end
|
||||
|
||||
# ## Получение корневого узла двоичного дерева ###
|
||||
def get_root
|
||||
@root
|
||||
end
|
||||
|
||||
# ## Получить высоту узла ###
|
||||
def height(node)
|
||||
# Высота пустого узла равна -1, высота листового узла равна 0
|
||||
return node.height unless node.nil?
|
||||
|
||||
-1
|
||||
end
|
||||
|
||||
# ## Обновить высоту узла ###
|
||||
### Обновить высоту узла ###
|
||||
def update_height(node)
|
||||
# Высота узла равна высоте более высокого поддерева + 1
|
||||
node.height = [height(node.left), height(node.right)].max + 1
|
||||
@@ -686,42 +636,7 @@ AVL-дерево одновременно является и двоичным
|
||||
=== "Ruby"
|
||||
|
||||
```ruby title="avl_tree.rb"
|
||||
=begin
|
||||
File: avl_tree.rb
|
||||
Created Time: 2024-04-17
|
||||
Author: Xuan Khoa Tu Nguyen (ngxktuzkai2000@gmail.com)
|
||||
=end
|
||||
|
||||
require_relative '../utils/tree_node'
|
||||
require_relative '../utils/print_util'
|
||||
|
||||
# ## AVL-дерево ###
|
||||
class AVLTree
|
||||
# ## Конструктор ###
|
||||
def initialize
|
||||
@root = nil
|
||||
end
|
||||
|
||||
# ## Получение корневого узла двоичного дерева ###
|
||||
def get_root
|
||||
@root
|
||||
end
|
||||
|
||||
# ## Получить высоту узла ###
|
||||
def height(node)
|
||||
# Высота пустого узла равна -1, высота листового узла равна 0
|
||||
return node.height unless node.nil?
|
||||
|
||||
-1
|
||||
end
|
||||
|
||||
# ## Обновить высоту узла ###
|
||||
def update_height(node)
|
||||
# Высота узла равна высоте более высокого поддерева + 1
|
||||
node.height = [height(node.left), height(node.right)].max + 1
|
||||
end
|
||||
|
||||
# ## Получить коэффициент баланса ###
|
||||
### Получить коэффициент баланса ###
|
||||
def balance_factor(node)
|
||||
# Коэффициент баланса пустого узла равен 0
|
||||
return 0 if node.nil?
|
||||
@@ -991,51 +906,7 @@ AVL-дерево одновременно является и двоичным
|
||||
=== "Ruby"
|
||||
|
||||
```ruby title="avl_tree.rb"
|
||||
=begin
|
||||
File: avl_tree.rb
|
||||
Created Time: 2024-04-17
|
||||
Author: Xuan Khoa Tu Nguyen (ngxktuzkai2000@gmail.com)
|
||||
=end
|
||||
|
||||
require_relative '../utils/tree_node'
|
||||
require_relative '../utils/print_util'
|
||||
|
||||
# ## AVL-дерево ###
|
||||
class AVLTree
|
||||
# ## Конструктор ###
|
||||
def initialize
|
||||
@root = nil
|
||||
end
|
||||
|
||||
# ## Получение корневого узла двоичного дерева ###
|
||||
def get_root
|
||||
@root
|
||||
end
|
||||
|
||||
# ## Получить высоту узла ###
|
||||
def height(node)
|
||||
# Высота пустого узла равна -1, высота листового узла равна 0
|
||||
return node.height unless node.nil?
|
||||
|
||||
-1
|
||||
end
|
||||
|
||||
# ## Обновить высоту узла ###
|
||||
def update_height(node)
|
||||
# Высота узла равна высоте более высокого поддерева + 1
|
||||
node.height = [height(node.left), height(node.right)].max + 1
|
||||
end
|
||||
|
||||
# ## Получить коэффициент баланса ###
|
||||
def balance_factor(node)
|
||||
# Коэффициент баланса пустого узла равен 0
|
||||
return 0 if node.nil?
|
||||
|
||||
# Коэффициент баланса узла = высота левого поддерева - высота правого поддерева
|
||||
height(node.left) - height(node.right)
|
||||
end
|
||||
|
||||
# ## Операция правого вращения ###
|
||||
### Операция правого вращения ###
|
||||
def right_rotate(node)
|
||||
child = node.left
|
||||
grand_child = child.right
|
||||
@@ -1290,65 +1161,7 @@ AVL-дерево одновременно является и двоичным
|
||||
=== "Ruby"
|
||||
|
||||
```ruby title="avl_tree.rb"
|
||||
=begin
|
||||
File: avl_tree.rb
|
||||
Created Time: 2024-04-17
|
||||
Author: Xuan Khoa Tu Nguyen (ngxktuzkai2000@gmail.com)
|
||||
=end
|
||||
|
||||
require_relative '../utils/tree_node'
|
||||
require_relative '../utils/print_util'
|
||||
|
||||
# ## AVL-дерево ###
|
||||
class AVLTree
|
||||
# ## Конструктор ###
|
||||
def initialize
|
||||
@root = nil
|
||||
end
|
||||
|
||||
# ## Получение корневого узла двоичного дерева ###
|
||||
def get_root
|
||||
@root
|
||||
end
|
||||
|
||||
# ## Получить высоту узла ###
|
||||
def height(node)
|
||||
# Высота пустого узла равна -1, высота листового узла равна 0
|
||||
return node.height unless node.nil?
|
||||
|
||||
-1
|
||||
end
|
||||
|
||||
# ## Обновить высоту узла ###
|
||||
def update_height(node)
|
||||
# Высота узла равна высоте более высокого поддерева + 1
|
||||
node.height = [height(node.left), height(node.right)].max + 1
|
||||
end
|
||||
|
||||
# ## Получить коэффициент баланса ###
|
||||
def balance_factor(node)
|
||||
# Коэффициент баланса пустого узла равен 0
|
||||
return 0 if node.nil?
|
||||
|
||||
# Коэффициент баланса узла = высота левого поддерева - высота правого поддерева
|
||||
height(node.left) - height(node.right)
|
||||
end
|
||||
|
||||
# ## Операция правого вращения ###
|
||||
def right_rotate(node)
|
||||
child = node.left
|
||||
grand_child = child.right
|
||||
# Выполнить правое вращение узла node вокруг child
|
||||
child.right = node
|
||||
node.left = grand_child
|
||||
# Обновить высоту узла
|
||||
update_height(node)
|
||||
update_height(child)
|
||||
# Вернуть корневой узел поддерева после вращения
|
||||
child
|
||||
end
|
||||
|
||||
# ## Операция левого вращения ###
|
||||
### Операция левого вращения ###
|
||||
def left_rotate(node)
|
||||
child = node.right
|
||||
grand_child = child.left
|
||||
@@ -1816,79 +1629,7 @@ AVL-дерево одновременно является и двоичным
|
||||
=== "Ruby"
|
||||
|
||||
```ruby title="avl_tree.rb"
|
||||
=begin
|
||||
File: avl_tree.rb
|
||||
Created Time: 2024-04-17
|
||||
Author: Xuan Khoa Tu Nguyen (ngxktuzkai2000@gmail.com)
|
||||
=end
|
||||
|
||||
require_relative '../utils/tree_node'
|
||||
require_relative '../utils/print_util'
|
||||
|
||||
# ## AVL-дерево ###
|
||||
class AVLTree
|
||||
# ## Конструктор ###
|
||||
def initialize
|
||||
@root = nil
|
||||
end
|
||||
|
||||
# ## Получение корневого узла двоичного дерева ###
|
||||
def get_root
|
||||
@root
|
||||
end
|
||||
|
||||
# ## Получить высоту узла ###
|
||||
def height(node)
|
||||
# Высота пустого узла равна -1, высота листового узла равна 0
|
||||
return node.height unless node.nil?
|
||||
|
||||
-1
|
||||
end
|
||||
|
||||
# ## Обновить высоту узла ###
|
||||
def update_height(node)
|
||||
# Высота узла равна высоте более высокого поддерева + 1
|
||||
node.height = [height(node.left), height(node.right)].max + 1
|
||||
end
|
||||
|
||||
# ## Получить коэффициент баланса ###
|
||||
def balance_factor(node)
|
||||
# Коэффициент баланса пустого узла равен 0
|
||||
return 0 if node.nil?
|
||||
|
||||
# Коэффициент баланса узла = высота левого поддерева - высота правого поддерева
|
||||
height(node.left) - height(node.right)
|
||||
end
|
||||
|
||||
# ## Операция правого вращения ###
|
||||
def right_rotate(node)
|
||||
child = node.left
|
||||
grand_child = child.right
|
||||
# Выполнить правое вращение узла node вокруг child
|
||||
child.right = node
|
||||
node.left = grand_child
|
||||
# Обновить высоту узла
|
||||
update_height(node)
|
||||
update_height(child)
|
||||
# Вернуть корневой узел поддерева после вращения
|
||||
child
|
||||
end
|
||||
|
||||
# ## Операция левого вращения ###
|
||||
def left_rotate(node)
|
||||
child = node.right
|
||||
grand_child = child.left
|
||||
# Выполнить левое вращение узла node вокруг child
|
||||
child.left = node
|
||||
node.right = grand_child
|
||||
# Обновить высоту узла
|
||||
update_height(node)
|
||||
update_height(child)
|
||||
# Вернуть корневой узел поддерева после вращения
|
||||
child
|
||||
end
|
||||
|
||||
# ## Выполнить вращение, чтобы снова сбалансировать поддерево ###
|
||||
### Выполнить вращение, чтобы снова сбалансировать поддерево ###
|
||||
def rotate(node)
|
||||
# Получить коэффициент баланса узла node
|
||||
balance_factor = balance_factor(node)
|
||||
@@ -2272,219 +2013,12 @@ AVL-дерево одновременно является и двоичным
|
||||
=== "Ruby"
|
||||
|
||||
```ruby title="avl_tree.rb"
|
||||
=begin
|
||||
File: avl_tree.rb
|
||||
Created Time: 2024-04-17
|
||||
Author: Xuan Khoa Tu Nguyen (ngxktuzkai2000@gmail.com)
|
||||
=end
|
||||
|
||||
require_relative '../utils/tree_node'
|
||||
require_relative '../utils/print_util'
|
||||
|
||||
# ## AVL-дерево ###
|
||||
class AVLTree
|
||||
# ## Конструктор ###
|
||||
def initialize
|
||||
@root = nil
|
||||
end
|
||||
|
||||
# ## Получение корневого узла двоичного дерева ###
|
||||
def get_root
|
||||
@root
|
||||
end
|
||||
|
||||
# ## Получить высоту узла ###
|
||||
def height(node)
|
||||
# Высота пустого узла равна -1, высота листового узла равна 0
|
||||
return node.height unless node.nil?
|
||||
|
||||
-1
|
||||
end
|
||||
|
||||
# ## Обновить высоту узла ###
|
||||
def update_height(node)
|
||||
# Высота узла равна высоте более высокого поддерева + 1
|
||||
node.height = [height(node.left), height(node.right)].max + 1
|
||||
end
|
||||
|
||||
# ## Получить коэффициент баланса ###
|
||||
def balance_factor(node)
|
||||
# Коэффициент баланса пустого узла равен 0
|
||||
return 0 if node.nil?
|
||||
|
||||
# Коэффициент баланса узла = высота левого поддерева - высота правого поддерева
|
||||
height(node.left) - height(node.right)
|
||||
end
|
||||
|
||||
# ## Операция правого вращения ###
|
||||
def right_rotate(node)
|
||||
child = node.left
|
||||
grand_child = child.right
|
||||
# Выполнить правое вращение узла node вокруг child
|
||||
child.right = node
|
||||
node.left = grand_child
|
||||
# Обновить высоту узла
|
||||
update_height(node)
|
||||
update_height(child)
|
||||
# Вернуть корневой узел поддерева после вращения
|
||||
child
|
||||
end
|
||||
|
||||
# ## Операция левого вращения ###
|
||||
def left_rotate(node)
|
||||
child = node.right
|
||||
grand_child = child.left
|
||||
# Выполнить левое вращение узла node вокруг child
|
||||
child.left = node
|
||||
node.right = grand_child
|
||||
# Обновить высоту узла
|
||||
update_height(node)
|
||||
update_height(child)
|
||||
# Вернуть корневой узел поддерева после вращения
|
||||
child
|
||||
end
|
||||
|
||||
# ## Выполнить вращение, чтобы снова сбалансировать поддерево ###
|
||||
def rotate(node)
|
||||
# Получить коэффициент баланса узла node
|
||||
balance_factor = balance_factor(node)
|
||||
# Обойти левое поддерево
|
||||
if balance_factor > 1
|
||||
if balance_factor(node.left) >= 0
|
||||
# Правое вращение
|
||||
return right_rotate(node)
|
||||
else
|
||||
# Сначала левое вращение, затем правое
|
||||
node.left = left_rotate(node.left)
|
||||
return right_rotate(node)
|
||||
end
|
||||
# Правостороннее дерево обхода
|
||||
elsif balance_factor < -1
|
||||
if balance_factor(node.right) <= 0
|
||||
# Левое вращение
|
||||
return left_rotate(node)
|
||||
else
|
||||
# Сначала правое вращение, затем левое
|
||||
node.right = right_rotate(node.right)
|
||||
return left_rotate(node)
|
||||
end
|
||||
end
|
||||
# Дерево сбалансировано, вращение не требуется, вернуть сразу
|
||||
node
|
||||
end
|
||||
|
||||
# ## Вставка узла ###
|
||||
### Вставка узла ###
|
||||
def insert(val)
|
||||
@root = insert_helper(@root, val)
|
||||
end
|
||||
|
||||
=begin
|
||||
File: avl_tree.rb
|
||||
Created Time: 2024-04-17
|
||||
Author: Xuan Khoa Tu Nguyen (ngxktuzkai2000@gmail.com)
|
||||
=end
|
||||
|
||||
require_relative '../utils/tree_node'
|
||||
require_relative '../utils/print_util'
|
||||
|
||||
# ## AVL-дерево ###
|
||||
class AVLTree
|
||||
# ## Конструктор ###
|
||||
def initialize
|
||||
@root = nil
|
||||
end
|
||||
|
||||
# ## Получение корневого узла двоичного дерева ###
|
||||
def get_root
|
||||
@root
|
||||
end
|
||||
|
||||
# ## Получить высоту узла ###
|
||||
def height(node)
|
||||
# Высота пустого узла равна -1, высота листового узла равна 0
|
||||
return node.height unless node.nil?
|
||||
|
||||
-1
|
||||
end
|
||||
|
||||
# ## Обновить высоту узла ###
|
||||
def update_height(node)
|
||||
# Высота узла равна высоте более высокого поддерева + 1
|
||||
node.height = [height(node.left), height(node.right)].max + 1
|
||||
end
|
||||
|
||||
# ## Получить коэффициент баланса ###
|
||||
def balance_factor(node)
|
||||
# Коэффициент баланса пустого узла равен 0
|
||||
return 0 if node.nil?
|
||||
|
||||
# Коэффициент баланса узла = высота левого поддерева - высота правого поддерева
|
||||
height(node.left) - height(node.right)
|
||||
end
|
||||
|
||||
# ## Операция правого вращения ###
|
||||
def right_rotate(node)
|
||||
child = node.left
|
||||
grand_child = child.right
|
||||
# Выполнить правое вращение узла node вокруг child
|
||||
child.right = node
|
||||
node.left = grand_child
|
||||
# Обновить высоту узла
|
||||
update_height(node)
|
||||
update_height(child)
|
||||
# Вернуть корневой узел поддерева после вращения
|
||||
child
|
||||
end
|
||||
|
||||
# ## Операция левого вращения ###
|
||||
def left_rotate(node)
|
||||
child = node.right
|
||||
grand_child = child.left
|
||||
# Выполнить левое вращение узла node вокруг child
|
||||
child.left = node
|
||||
node.right = grand_child
|
||||
# Обновить высоту узла
|
||||
update_height(node)
|
||||
update_height(child)
|
||||
# Вернуть корневой узел поддерева после вращения
|
||||
child
|
||||
end
|
||||
|
||||
# ## Выполнить вращение, чтобы снова сбалансировать поддерево ###
|
||||
def rotate(node)
|
||||
# Получить коэффициент баланса узла node
|
||||
balance_factor = balance_factor(node)
|
||||
# Обойти левое поддерево
|
||||
if balance_factor > 1
|
||||
if balance_factor(node.left) >= 0
|
||||
# Правое вращение
|
||||
return right_rotate(node)
|
||||
else
|
||||
# Сначала левое вращение, затем правое
|
||||
node.left = left_rotate(node.left)
|
||||
return right_rotate(node)
|
||||
end
|
||||
# Правостороннее дерево обхода
|
||||
elsif balance_factor < -1
|
||||
if balance_factor(node.right) <= 0
|
||||
# Левое вращение
|
||||
return left_rotate(node)
|
||||
else
|
||||
# Сначала правое вращение, затем левое
|
||||
node.right = right_rotate(node.right)
|
||||
return left_rotate(node)
|
||||
end
|
||||
end
|
||||
# Дерево сбалансировано, вращение не требуется, вернуть сразу
|
||||
node
|
||||
end
|
||||
|
||||
# ## Вставка узла ###
|
||||
def insert(val)
|
||||
@root = insert_helper(@root, val)
|
||||
end
|
||||
|
||||
# ## Рекурсивная вставка узла (вспомогательный метод) ###
|
||||
### Рекурсивная вставка узла (вспомогательный метод) ###
|
||||
def insert_helper(node, val)
|
||||
return TreeNode.new(val) if node.nil?
|
||||
# 1. Найти позицию вставки и вставить узел
|
||||
@@ -3067,265 +2601,12 @@ AVL-дерево одновременно является и двоичным
|
||||
=== "Ruby"
|
||||
|
||||
```ruby title="avl_tree.rb"
|
||||
=begin
|
||||
File: avl_tree.rb
|
||||
Created Time: 2024-04-17
|
||||
Author: Xuan Khoa Tu Nguyen (ngxktuzkai2000@gmail.com)
|
||||
=end
|
||||
|
||||
require_relative '../utils/tree_node'
|
||||
require_relative '../utils/print_util'
|
||||
|
||||
# ## AVL-дерево ###
|
||||
class AVLTree
|
||||
# ## Конструктор ###
|
||||
def initialize
|
||||
@root = nil
|
||||
end
|
||||
|
||||
# ## Получение корневого узла двоичного дерева ###
|
||||
def get_root
|
||||
@root
|
||||
end
|
||||
|
||||
# ## Получить высоту узла ###
|
||||
def height(node)
|
||||
# Высота пустого узла равна -1, высота листового узла равна 0
|
||||
return node.height unless node.nil?
|
||||
|
||||
-1
|
||||
end
|
||||
|
||||
# ## Обновить высоту узла ###
|
||||
def update_height(node)
|
||||
# Высота узла равна высоте более высокого поддерева + 1
|
||||
node.height = [height(node.left), height(node.right)].max + 1
|
||||
end
|
||||
|
||||
# ## Получить коэффициент баланса ###
|
||||
def balance_factor(node)
|
||||
# Коэффициент баланса пустого узла равен 0
|
||||
return 0 if node.nil?
|
||||
|
||||
# Коэффициент баланса узла = высота левого поддерева - высота правого поддерева
|
||||
height(node.left) - height(node.right)
|
||||
end
|
||||
|
||||
# ## Операция правого вращения ###
|
||||
def right_rotate(node)
|
||||
child = node.left
|
||||
grand_child = child.right
|
||||
# Выполнить правое вращение узла node вокруг child
|
||||
child.right = node
|
||||
node.left = grand_child
|
||||
# Обновить высоту узла
|
||||
update_height(node)
|
||||
update_height(child)
|
||||
# Вернуть корневой узел поддерева после вращения
|
||||
child
|
||||
end
|
||||
|
||||
# ## Операция левого вращения ###
|
||||
def left_rotate(node)
|
||||
child = node.right
|
||||
grand_child = child.left
|
||||
# Выполнить левое вращение узла node вокруг child
|
||||
child.left = node
|
||||
node.right = grand_child
|
||||
# Обновить высоту узла
|
||||
update_height(node)
|
||||
update_height(child)
|
||||
# Вернуть корневой узел поддерева после вращения
|
||||
child
|
||||
end
|
||||
|
||||
# ## Выполнить вращение, чтобы снова сбалансировать поддерево ###
|
||||
def rotate(node)
|
||||
# Получить коэффициент баланса узла node
|
||||
balance_factor = balance_factor(node)
|
||||
# Обойти левое поддерево
|
||||
if balance_factor > 1
|
||||
if balance_factor(node.left) >= 0
|
||||
# Правое вращение
|
||||
return right_rotate(node)
|
||||
else
|
||||
# Сначала левое вращение, затем правое
|
||||
node.left = left_rotate(node.left)
|
||||
return right_rotate(node)
|
||||
end
|
||||
# Правостороннее дерево обхода
|
||||
elsif balance_factor < -1
|
||||
if balance_factor(node.right) <= 0
|
||||
# Левое вращение
|
||||
return left_rotate(node)
|
||||
else
|
||||
# Сначала правое вращение, затем левое
|
||||
node.right = right_rotate(node.right)
|
||||
return left_rotate(node)
|
||||
end
|
||||
end
|
||||
# Дерево сбалансировано, вращение не требуется, вернуть сразу
|
||||
node
|
||||
end
|
||||
|
||||
# ## Вставка узла ###
|
||||
def insert(val)
|
||||
@root = insert_helper(@root, val)
|
||||
end
|
||||
|
||||
# ## Рекурсивная вставка узла (вспомогательный метод) ###
|
||||
def insert_helper(node, val)
|
||||
return TreeNode.new(val) if node.nil?
|
||||
# 1. Найти позицию вставки и вставить узел
|
||||
if val < node.val
|
||||
node.left = insert_helper(node.left, val)
|
||||
elsif val > node.val
|
||||
node.right = insert_helper(node.right, val)
|
||||
else
|
||||
# Повторяющийся узел не вставлять, сразу вернуть
|
||||
return node
|
||||
end
|
||||
# Обновить высоту узла
|
||||
update_height(node)
|
||||
# 2. Выполнить вращение, чтобы снова сбалансировать поддерево
|
||||
rotate(node)
|
||||
end
|
||||
|
||||
# ## Удаление узла ###
|
||||
### Удаление узла ###
|
||||
def remove(val)
|
||||
@root = remove_helper(@root, val)
|
||||
end
|
||||
|
||||
=begin
|
||||
File: avl_tree.rb
|
||||
Created Time: 2024-04-17
|
||||
Author: Xuan Khoa Tu Nguyen (ngxktuzkai2000@gmail.com)
|
||||
=end
|
||||
|
||||
require_relative '../utils/tree_node'
|
||||
require_relative '../utils/print_util'
|
||||
|
||||
# ## AVL-дерево ###
|
||||
class AVLTree
|
||||
# ## Конструктор ###
|
||||
def initialize
|
||||
@root = nil
|
||||
end
|
||||
|
||||
# ## Получение корневого узла двоичного дерева ###
|
||||
def get_root
|
||||
@root
|
||||
end
|
||||
|
||||
# ## Получить высоту узла ###
|
||||
def height(node)
|
||||
# Высота пустого узла равна -1, высота листового узла равна 0
|
||||
return node.height unless node.nil?
|
||||
|
||||
-1
|
||||
end
|
||||
|
||||
# ## Обновить высоту узла ###
|
||||
def update_height(node)
|
||||
# Высота узла равна высоте более высокого поддерева + 1
|
||||
node.height = [height(node.left), height(node.right)].max + 1
|
||||
end
|
||||
|
||||
# ## Получить коэффициент баланса ###
|
||||
def balance_factor(node)
|
||||
# Коэффициент баланса пустого узла равен 0
|
||||
return 0 if node.nil?
|
||||
|
||||
# Коэффициент баланса узла = высота левого поддерева - высота правого поддерева
|
||||
height(node.left) - height(node.right)
|
||||
end
|
||||
|
||||
# ## Операция правого вращения ###
|
||||
def right_rotate(node)
|
||||
child = node.left
|
||||
grand_child = child.right
|
||||
# Выполнить правое вращение узла node вокруг child
|
||||
child.right = node
|
||||
node.left = grand_child
|
||||
# Обновить высоту узла
|
||||
update_height(node)
|
||||
update_height(child)
|
||||
# Вернуть корневой узел поддерева после вращения
|
||||
child
|
||||
end
|
||||
|
||||
# ## Операция левого вращения ###
|
||||
def left_rotate(node)
|
||||
child = node.right
|
||||
grand_child = child.left
|
||||
# Выполнить левое вращение узла node вокруг child
|
||||
child.left = node
|
||||
node.right = grand_child
|
||||
# Обновить высоту узла
|
||||
update_height(node)
|
||||
update_height(child)
|
||||
# Вернуть корневой узел поддерева после вращения
|
||||
child
|
||||
end
|
||||
|
||||
# ## Выполнить вращение, чтобы снова сбалансировать поддерево ###
|
||||
def rotate(node)
|
||||
# Получить коэффициент баланса узла node
|
||||
balance_factor = balance_factor(node)
|
||||
# Обойти левое поддерево
|
||||
if balance_factor > 1
|
||||
if balance_factor(node.left) >= 0
|
||||
# Правое вращение
|
||||
return right_rotate(node)
|
||||
else
|
||||
# Сначала левое вращение, затем правое
|
||||
node.left = left_rotate(node.left)
|
||||
return right_rotate(node)
|
||||
end
|
||||
# Правостороннее дерево обхода
|
||||
elsif balance_factor < -1
|
||||
if balance_factor(node.right) <= 0
|
||||
# Левое вращение
|
||||
return left_rotate(node)
|
||||
else
|
||||
# Сначала правое вращение, затем левое
|
||||
node.right = right_rotate(node.right)
|
||||
return left_rotate(node)
|
||||
end
|
||||
end
|
||||
# Дерево сбалансировано, вращение не требуется, вернуть сразу
|
||||
node
|
||||
end
|
||||
|
||||
# ## Вставка узла ###
|
||||
def insert(val)
|
||||
@root = insert_helper(@root, val)
|
||||
end
|
||||
|
||||
# ## Рекурсивная вставка узла (вспомогательный метод) ###
|
||||
def insert_helper(node, val)
|
||||
return TreeNode.new(val) if node.nil?
|
||||
# 1. Найти позицию вставки и вставить узел
|
||||
if val < node.val
|
||||
node.left = insert_helper(node.left, val)
|
||||
elsif val > node.val
|
||||
node.right = insert_helper(node.right, val)
|
||||
else
|
||||
# Повторяющийся узел не вставлять, сразу вернуть
|
||||
return node
|
||||
end
|
||||
# Обновить высоту узла
|
||||
update_height(node)
|
||||
# 2. Выполнить вращение, чтобы снова сбалансировать поддерево
|
||||
rotate(node)
|
||||
end
|
||||
|
||||
# ## Удаление узла ###
|
||||
def remove(val)
|
||||
@root = remove_helper(@root, val)
|
||||
end
|
||||
|
||||
# ## Рекурсивное удаление узла (вспомогательный метод) ###
|
||||
### Рекурсивное удаление узла (вспомогательный метод) ###
|
||||
def remove_helper(node, val)
|
||||
return if node.nil?
|
||||
# 1. Найти узел и удалить его
|
||||
|
||||
Reference in New Issue
Block a user