mirror of
https://github.com/krahets/hello-algo.git
synced 2026-09-17 20:17:14 +00:00
Add ru version (#1865)
* Add Russian docs site baseline * Add Russian localized codebase * Polish Russian code wording * Update ru code translation. * Update code translation and chapter covers. * Fix pythontutor extraction. * Add README and landing page. * placeholder of profiles * Use figures of English version * Remove chapter paperbook
This commit is contained in:
@@ -0,0 +1,62 @@
|
||||
/**
|
||||
* File: heap.swift
|
||||
* Created Time: 2024-03-17
|
||||
* Author: nuomi1 (nuomi1@qq.com)
|
||||
*/
|
||||
|
||||
import HeapModule
|
||||
import utils
|
||||
|
||||
func testPush(heap: inout Heap<Int>, val: Int) {
|
||||
heap.insert(val)
|
||||
print("\nПосле добавления элемента \(val) в кучу\n")
|
||||
PrintUtil.printHeap(queue: heap.unordered)
|
||||
}
|
||||
|
||||
func testPop(heap: inout Heap<Int>) {
|
||||
let val = heap.removeMax()
|
||||
print("\nПосле извлечения элемента вершины кучи \(val)\n")
|
||||
PrintUtil.printHeap(queue: heap.unordered)
|
||||
}
|
||||
|
||||
@main
|
||||
enum _Heap {
|
||||
/* Driver Code */
|
||||
static func main() {
|
||||
/* Инициализация кучи */
|
||||
// Тип Heap в Swift одновременно поддерживает максимальную и минимальную кучу
|
||||
var heap = Heap<Int>()
|
||||
|
||||
/* Добавление элемента в кучу */
|
||||
testPush(heap: &heap, val: 1)
|
||||
testPush(heap: &heap, val: 3)
|
||||
testPush(heap: &heap, val: 2)
|
||||
testPush(heap: &heap, val: 5)
|
||||
testPush(heap: &heap, val: 4)
|
||||
|
||||
/* Получение элемента с вершины кучи */
|
||||
let peek = heap.max()
|
||||
print("\nЭлемент на вершине кучи = \(peek!)\n")
|
||||
|
||||
/* Извлечение элемента с вершины кучи */
|
||||
testPop(heap: &heap)
|
||||
testPop(heap: &heap)
|
||||
testPop(heap: &heap)
|
||||
testPop(heap: &heap)
|
||||
testPop(heap: &heap)
|
||||
|
||||
/* Получение размера кучи */
|
||||
let size = heap.count
|
||||
print("\nКоличество элементов в куче = \(size)\n")
|
||||
|
||||
/* Проверка, пуста ли куча */
|
||||
let isEmpty = heap.isEmpty
|
||||
print("\nПуста ли куча: \(isEmpty)\n")
|
||||
|
||||
/* Построить кучу по входному списку */
|
||||
// Временная сложность равна O(n), а не O(nlogn)
|
||||
let heap2 = Heap([1, 3, 2, 5, 4])
|
||||
print("\nПосле построения кучи из входного списка")
|
||||
PrintUtil.printHeap(queue: heap2.unordered)
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,163 @@
|
||||
/**
|
||||
* File: my_heap.swift
|
||||
* Created Time: 2023-01-28
|
||||
* Author: nuomi1 (nuomi1@qq.com)
|
||||
*/
|
||||
|
||||
import utils
|
||||
|
||||
/* Максимальная куча */
|
||||
class MaxHeap {
|
||||
private var maxHeap: [Int]
|
||||
|
||||
/* Конструктор, строящий кучу по входному списку */
|
||||
init(nums: [Int]) {
|
||||
// Добавить элементы списка в кучу без изменений
|
||||
maxHeap = nums
|
||||
// Выполнить heapify для всех узлов, кроме листовых
|
||||
for i in (0 ... parent(i: size() - 1)).reversed() {
|
||||
siftDown(i: i)
|
||||
}
|
||||
}
|
||||
|
||||
/* Получить индекс левого дочернего узла */
|
||||
private func left(i: Int) -> Int {
|
||||
2 * i + 1
|
||||
}
|
||||
|
||||
/* Получить индекс правого дочернего узла */
|
||||
private func right(i: Int) -> Int {
|
||||
2 * i + 2
|
||||
}
|
||||
|
||||
/* Получить индекс родительского узла */
|
||||
private func parent(i: Int) -> Int {
|
||||
(i - 1) / 2 // Округление вниз при делении
|
||||
}
|
||||
|
||||
/* Поменять элементы местами */
|
||||
private func swap(i: Int, j: Int) {
|
||||
maxHeap.swapAt(i, j)
|
||||
}
|
||||
|
||||
/* Получение размера кучи */
|
||||
func size() -> Int {
|
||||
maxHeap.count
|
||||
}
|
||||
|
||||
/* Проверка, пуста ли куча */
|
||||
func isEmpty() -> Bool {
|
||||
size() == 0
|
||||
}
|
||||
|
||||
/* Доступ к элементу на вершине кучи */
|
||||
func peek() -> Int {
|
||||
maxHeap[0]
|
||||
}
|
||||
|
||||
/* Добавление элемента в кучу */
|
||||
func push(val: Int) {
|
||||
// Добавление узла
|
||||
maxHeap.append(val)
|
||||
// Просеивание снизу вверх
|
||||
siftUp(i: size() - 1)
|
||||
}
|
||||
|
||||
/* Начиная с узла i, выполнить просеивание снизу вверх */
|
||||
private func siftUp(i: Int) {
|
||||
var i = i
|
||||
while true {
|
||||
// Получение родительского узла для узла i
|
||||
let p = parent(i: i)
|
||||
// Завершить heapify, когда «корневой узел уже пройден» или «узел не требует исправления»
|
||||
if p < 0 || maxHeap[i] <= maxHeap[p] {
|
||||
break
|
||||
}
|
||||
// Поменять два узла местами
|
||||
swap(i: i, j: p)
|
||||
// Циклическое просеивание вверх
|
||||
i = p
|
||||
}
|
||||
}
|
||||
|
||||
/* Извлечение элемента из кучи */
|
||||
func pop() -> Int {
|
||||
// Обработка пустого случая
|
||||
if isEmpty() {
|
||||
fatalError("куча пуста")
|
||||
}
|
||||
// Поменять корневой узел с самым правым листом местами (поменять первый и последний элементы)
|
||||
swap(i: 0, j: size() - 1)
|
||||
// Удаление узла
|
||||
let val = maxHeap.remove(at: size() - 1)
|
||||
// Просеивание сверху вниз
|
||||
siftDown(i: 0)
|
||||
// Вернуть элемент с вершины кучи
|
||||
return val
|
||||
}
|
||||
|
||||
/* Начиная с узла i, выполнить просеивание сверху вниз */
|
||||
private func siftDown(i: Int) {
|
||||
var i = i
|
||||
while true {
|
||||
// Определить узел с максимальным значением среди i, l и r и обозначить его как ma
|
||||
let l = left(i: i)
|
||||
let r = right(i: i)
|
||||
var ma = i
|
||||
if l < size(), maxHeap[l] > maxHeap[ma] {
|
||||
ma = l
|
||||
}
|
||||
if r < size(), maxHeap[r] > maxHeap[ma] {
|
||||
ma = r
|
||||
}
|
||||
// Если узел i уже максимален или индексы l и r вне границ, дальнейшее просеивание не требуется, выйти
|
||||
if ma == i {
|
||||
break
|
||||
}
|
||||
// Поменять два узла местами
|
||||
swap(i: i, j: ma)
|
||||
// Циклическое просеивание вниз
|
||||
i = ma
|
||||
}
|
||||
}
|
||||
|
||||
/* Вывести кучу (двоичное дерево) */
|
||||
func print() {
|
||||
let queue = maxHeap
|
||||
PrintUtil.printHeap(queue: queue)
|
||||
}
|
||||
}
|
||||
|
||||
@main
|
||||
enum MyHeap {
|
||||
/* Driver Code */
|
||||
static func main() {
|
||||
/* Инициализация максимальной кучи */
|
||||
let maxHeap = MaxHeap(nums: [9, 8, 6, 6, 7, 5, 2, 1, 4, 3, 6, 2])
|
||||
print("\nПосле построения кучи из входного списка")
|
||||
maxHeap.print()
|
||||
|
||||
/* Получение элемента с вершины кучи */
|
||||
var peek = maxHeap.peek()
|
||||
print("\nЭлемент на вершине кучи = \(peek)")
|
||||
|
||||
/* Добавление элемента в кучу */
|
||||
let val = 7
|
||||
maxHeap.push(val: val)
|
||||
print("\nПосле добавления элемента \(val) в кучу")
|
||||
maxHeap.print()
|
||||
|
||||
/* Извлечение элемента с вершины кучи */
|
||||
peek = maxHeap.pop()
|
||||
print("\nПосле извлечения элемента вершины кучи \(peek)")
|
||||
maxHeap.print()
|
||||
|
||||
/* Получение размера кучи */
|
||||
let size = maxHeap.size()
|
||||
print("\nКоличество элементов в куче = \(size)")
|
||||
|
||||
/* Проверка, пуста ли куча */
|
||||
let isEmpty = maxHeap.isEmpty()
|
||||
print("\nПуста ли куча: \(isEmpty)")
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,36 @@
|
||||
/**
|
||||
* File: top_k.swift
|
||||
* Created Time: 2023-07-02
|
||||
* Author: nuomi1 (nuomi1@qq.com)
|
||||
*/
|
||||
|
||||
import HeapModule
|
||||
import utils
|
||||
|
||||
/* Найти k наибольших элементов массива с помощью кучи */
|
||||
func topKHeap(nums: [Int], k: Int) -> [Int] {
|
||||
// Инициализировать минимальную кучу и построить ее по первым k элементам
|
||||
var heap = Heap(nums.prefix(k))
|
||||
// Начиная с элемента k+1, поддерживать длину кучи равной k
|
||||
for i in nums.indices.dropFirst(k) {
|
||||
// Если текущий элемент больше элемента на вершине кучи, извлечь вершину кучи и добавить текущий элемент в кучу
|
||||
if nums[i] > heap.min()! {
|
||||
_ = heap.removeMin()
|
||||
heap.insert(nums[i])
|
||||
}
|
||||
}
|
||||
return heap.unordered
|
||||
}
|
||||
|
||||
@main
|
||||
enum TopK {
|
||||
/* Driver Code */
|
||||
static func main() {
|
||||
let nums = [1, 7, 6, 3, 2]
|
||||
let k = 3
|
||||
|
||||
let res = topKHeap(nums: nums, k: k)
|
||||
print("Наибольшие \(k) элементов")
|
||||
PrintUtil.printHeap(queue: res)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user