mirror of
https://github.com/krahets/hello-algo.git
synced 2026-07-07 11:36:46 +08:00
build
This commit is contained in:
@@ -1245,28 +1245,19 @@ comments: true
|
||||
=== "Ruby"
|
||||
|
||||
```ruby title="array_binary_tree.rb"
|
||||
=begin
|
||||
File: array_binary_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'
|
||||
|
||||
# ## Класс двоичного дерева в массивном представлении ###
|
||||
### Класс двоичного дерева в массивном представлении ###
|
||||
class ArrayBinaryTree
|
||||
# ## Конструктор ###
|
||||
### Конструктор ###
|
||||
def initialize(arr)
|
||||
@tree = arr.to_a
|
||||
end
|
||||
|
||||
# ## Вместимость списка ###
|
||||
### Вместимость списка ###
|
||||
def size
|
||||
@tree.length
|
||||
end
|
||||
|
||||
# ## Получить значение узла с индексом i ###
|
||||
### Получить значение узла с индексом i ###
|
||||
def val(i)
|
||||
# Если индекс выходит за границы, вернуть nil, обозначающий пустую ячейку
|
||||
return if i < 0 || i >= size
|
||||
@@ -1274,22 +1265,22 @@ comments: true
|
||||
@tree[i]
|
||||
end
|
||||
|
||||
# ## Получить индекс левого дочернего узла узла с индексом i ###
|
||||
### Получить индекс левого дочернего узла узла с индексом i ###
|
||||
def left(i)
|
||||
2 * i + 1
|
||||
end
|
||||
|
||||
# ## Получить индекс правого дочернего узла узла с индексом i ###
|
||||
### Получить индекс правого дочернего узла узла с индексом i ###
|
||||
def right(i)
|
||||
2 * i + 2
|
||||
end
|
||||
|
||||
# ## Получить индекс родительского узла узла с индексом i ###
|
||||
### Получить индекс родительского узла узла с индексом i ###
|
||||
def parent(i)
|
||||
(i - 1) / 2
|
||||
end
|
||||
|
||||
# ## Обход в ширину ###
|
||||
### Обход в ширину ###
|
||||
def level_order
|
||||
@res = []
|
||||
|
||||
@@ -1301,7 +1292,7 @@ comments: true
|
||||
@res
|
||||
end
|
||||
|
||||
# ## Обход в глубину ###
|
||||
### Обход в глубину ###
|
||||
def dfs(i, order)
|
||||
return if val(i).nil?
|
||||
# Предварительный обход
|
||||
@@ -1314,21 +1305,21 @@ comments: true
|
||||
@res << val(i) if order == :post
|
||||
end
|
||||
|
||||
# ## Предварительный обход ###
|
||||
### Предварительный обход ###
|
||||
def pre_order
|
||||
@res = []
|
||||
dfs(0, :pre)
|
||||
@res
|
||||
end
|
||||
|
||||
# ## Симметричный обход ###
|
||||
### Симметричный обход ###
|
||||
def in_order
|
||||
@res = []
|
||||
dfs(0, :in)
|
||||
@res
|
||||
end
|
||||
|
||||
# ## Обратный обход ###
|
||||
### Обратный обход ###
|
||||
def post_order
|
||||
@res = []
|
||||
dfs(0, :post)
|
||||
|
||||
@@ -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. Найти узел и удалить его
|
||||
|
||||
@@ -316,29 +316,7 @@ comments: true
|
||||
=== "Ruby"
|
||||
|
||||
```ruby title="binary_search_tree.rb"
|
||||
=begin
|
||||
File: binary_search_tree.rb
|
||||
Created Time: 2024-04-18
|
||||
Author: Xuan Khoa Tu Nguyen (ngxktuzkai2000@gmail.com)
|
||||
=end
|
||||
|
||||
require_relative '../utils/tree_node'
|
||||
require_relative '../utils/print_util'
|
||||
|
||||
# ## Двоичное дерево поиска ###
|
||||
class BinarySearchTree
|
||||
# ## Конструктор ###
|
||||
def initialize
|
||||
# Инициализировать пустое дерево
|
||||
@root = nil
|
||||
end
|
||||
|
||||
# ## Получение корневого узла двоичного дерева ###
|
||||
def get_root
|
||||
@root
|
||||
end
|
||||
|
||||
# ## Поиск узла ###
|
||||
### Поиск узла ###
|
||||
def search(num)
|
||||
cur = @root
|
||||
|
||||
@@ -790,50 +768,7 @@ comments: true
|
||||
=== "Ruby"
|
||||
|
||||
```ruby title="binary_search_tree.rb"
|
||||
=begin
|
||||
File: binary_search_tree.rb
|
||||
Created Time: 2024-04-18
|
||||
Author: Xuan Khoa Tu Nguyen (ngxktuzkai2000@gmail.com)
|
||||
=end
|
||||
|
||||
require_relative '../utils/tree_node'
|
||||
require_relative '../utils/print_util'
|
||||
|
||||
# ## Двоичное дерево поиска ###
|
||||
class BinarySearchTree
|
||||
# ## Конструктор ###
|
||||
def initialize
|
||||
# Инициализировать пустое дерево
|
||||
@root = nil
|
||||
end
|
||||
|
||||
# ## Получение корневого узла двоичного дерева ###
|
||||
def get_root
|
||||
@root
|
||||
end
|
||||
|
||||
# ## Поиск узла ###
|
||||
def search(num)
|
||||
cur = @root
|
||||
|
||||
# Искать в цикле и выйти после прохода за листовой узел
|
||||
while !cur.nil?
|
||||
# Целевой узел находится в правом поддереве cur
|
||||
if cur.val < num
|
||||
cur = cur.right
|
||||
# Целевой узел находится в левом поддереве cur
|
||||
elsif cur.val > num
|
||||
cur = cur.left
|
||||
# Найти целевой узел и выйти из цикла
|
||||
else
|
||||
break
|
||||
end
|
||||
end
|
||||
|
||||
cur
|
||||
end
|
||||
|
||||
# ## Вставка узла ###
|
||||
### Вставка узла ###
|
||||
def insert(num)
|
||||
# Если дерево пусто, инициализировать корневой узел
|
||||
if @root.nil?
|
||||
@@ -1600,83 +1535,7 @@ comments: true
|
||||
=== "Ruby"
|
||||
|
||||
```ruby title="binary_search_tree.rb"
|
||||
=begin
|
||||
File: binary_search_tree.rb
|
||||
Created Time: 2024-04-18
|
||||
Author: Xuan Khoa Tu Nguyen (ngxktuzkai2000@gmail.com)
|
||||
=end
|
||||
|
||||
require_relative '../utils/tree_node'
|
||||
require_relative '../utils/print_util'
|
||||
|
||||
# ## Двоичное дерево поиска ###
|
||||
class BinarySearchTree
|
||||
# ## Конструктор ###
|
||||
def initialize
|
||||
# Инициализировать пустое дерево
|
||||
@root = nil
|
||||
end
|
||||
|
||||
# ## Получение корневого узла двоичного дерева ###
|
||||
def get_root
|
||||
@root
|
||||
end
|
||||
|
||||
# ## Поиск узла ###
|
||||
def search(num)
|
||||
cur = @root
|
||||
|
||||
# Искать в цикле и выйти после прохода за листовой узел
|
||||
while !cur.nil?
|
||||
# Целевой узел находится в правом поддереве cur
|
||||
if cur.val < num
|
||||
cur = cur.right
|
||||
# Целевой узел находится в левом поддереве cur
|
||||
elsif cur.val > num
|
||||
cur = cur.left
|
||||
# Найти целевой узел и выйти из цикла
|
||||
else
|
||||
break
|
||||
end
|
||||
end
|
||||
|
||||
cur
|
||||
end
|
||||
|
||||
# ## Вставка узла ###
|
||||
def insert(num)
|
||||
# Если дерево пусто, инициализировать корневой узел
|
||||
if @root.nil?
|
||||
@root = TreeNode.new(num)
|
||||
return
|
||||
end
|
||||
|
||||
# Искать в цикле и выйти после прохода за листовой узел
|
||||
cur, pre = @root, nil
|
||||
while !cur.nil?
|
||||
# Найти повторяющийся узел и сразу вернуть
|
||||
return if cur.val == num
|
||||
|
||||
pre = cur
|
||||
# Позиция вставки находится в правом поддереве cur
|
||||
if cur.val < num
|
||||
cur = cur.right
|
||||
# Позиция вставки находится в левом поддереве cur
|
||||
else
|
||||
cur = cur.left
|
||||
end
|
||||
end
|
||||
|
||||
# Вставка узла
|
||||
node = TreeNode.new(num)
|
||||
if pre.val < num
|
||||
pre.right = node
|
||||
else
|
||||
pre.left = node
|
||||
end
|
||||
end
|
||||
|
||||
# ## Удаление узла ###
|
||||
### Удаление узла ###
|
||||
def remove(num)
|
||||
# Если дерево пусто, сразу вернуть
|
||||
return if @root.nil?
|
||||
|
||||
@@ -318,16 +318,7 @@ comments: true
|
||||
=== "Ruby"
|
||||
|
||||
```ruby title="binary_tree_bfs.rb"
|
||||
=begin
|
||||
File: binary_tree_bfs.rb
|
||||
Created Time: 2024-04-18
|
||||
Author: Xuan Khoa Tu Nguyen (ngxktuzkai2000@gmail.com)
|
||||
=end
|
||||
|
||||
require_relative '../utils/tree_node'
|
||||
require_relative '../utils/print_util'
|
||||
|
||||
# ## Обход в ширину ###
|
||||
### Обход в ширину ###
|
||||
def level_order(root)
|
||||
# Инициализировать очередь и добавить корневой узел
|
||||
queue = [root]
|
||||
@@ -797,16 +788,7 @@ comments: true
|
||||
=== "Ruby"
|
||||
|
||||
```ruby title="binary_tree_dfs.rb"
|
||||
=begin
|
||||
File: binary_tree_dfs.rb
|
||||
Created Time: 2024-04-18
|
||||
Author: Xuan Khoa Tu Nguyen (ngxktuzkai2000@gmail.com)
|
||||
=end
|
||||
|
||||
require_relative '../utils/tree_node'
|
||||
require_relative '../utils/print_util'
|
||||
|
||||
# ## Предварительный обход ###
|
||||
### Предварительный обход ###
|
||||
def pre_order(root)
|
||||
return if root.nil?
|
||||
|
||||
@@ -816,26 +798,7 @@ comments: true
|
||||
pre_order(root.right)
|
||||
end
|
||||
|
||||
=begin
|
||||
File: binary_tree_dfs.rb
|
||||
Created Time: 2024-04-18
|
||||
Author: Xuan Khoa Tu Nguyen (ngxktuzkai2000@gmail.com)
|
||||
=end
|
||||
|
||||
require_relative '../utils/tree_node'
|
||||
require_relative '../utils/print_util'
|
||||
|
||||
# ## Предварительный обход ###
|
||||
def pre_order(root)
|
||||
return if root.nil?
|
||||
|
||||
# Порядок обхода: корень -> левое поддерево -> правое поддерево
|
||||
$res << root.val
|
||||
pre_order(root.left)
|
||||
pre_order(root.right)
|
||||
end
|
||||
|
||||
# ## Симметричный обход ###
|
||||
### Симметричный обход ###
|
||||
def in_order(root)
|
||||
return if root.nil?
|
||||
|
||||
@@ -845,36 +808,7 @@ comments: true
|
||||
in_order(root.right)
|
||||
end
|
||||
|
||||
=begin
|
||||
File: binary_tree_dfs.rb
|
||||
Created Time: 2024-04-18
|
||||
Author: Xuan Khoa Tu Nguyen (ngxktuzkai2000@gmail.com)
|
||||
=end
|
||||
|
||||
require_relative '../utils/tree_node'
|
||||
require_relative '../utils/print_util'
|
||||
|
||||
# ## Предварительный обход ###
|
||||
def pre_order(root)
|
||||
return if root.nil?
|
||||
|
||||
# Порядок обхода: корень -> левое поддерево -> правое поддерево
|
||||
$res << root.val
|
||||
pre_order(root.left)
|
||||
pre_order(root.right)
|
||||
end
|
||||
|
||||
# ## Симметричный обход ###
|
||||
def in_order(root)
|
||||
return if root.nil?
|
||||
|
||||
# Порядок обхода: левое поддерево -> корень -> правое поддерево
|
||||
in_order(root.left)
|
||||
$res << root.val
|
||||
in_order(root.right)
|
||||
end
|
||||
|
||||
# ## Обратный обход ###
|
||||
### Обратный обход ###
|
||||
def post_order(root)
|
||||
return if root.nil?
|
||||
|
||||
|
||||
Reference in New Issue
Block a user