mirror of
https://github.com/krahets/hello-algo.git
synced 2026-07-22 19:36:06 +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,58 @@
|
||||
/**
|
||||
* File: top_k.js
|
||||
* Created Time: 2023-08-13
|
||||
* Author: Justin (xiefahit@gmail.com)
|
||||
*/
|
||||
|
||||
const { MaxHeap } = require('./my_heap');
|
||||
|
||||
/* Добавление элемента в кучу */
|
||||
function pushMinHeap(maxHeap, val) {
|
||||
// Инвертировать знак элемента
|
||||
maxHeap.push(-val);
|
||||
}
|
||||
|
||||
/* Извлечение элемента из кучи */
|
||||
function popMinHeap(maxHeap) {
|
||||
// Инвертировать знак элемента
|
||||
return -maxHeap.pop();
|
||||
}
|
||||
|
||||
/* Доступ к элементу на вершине кучи */
|
||||
function peekMinHeap(maxHeap) {
|
||||
// Инвертировать знак элемента
|
||||
return -maxHeap.peek();
|
||||
}
|
||||
|
||||
/* Извлечь элементы из кучи */
|
||||
function getMinHeap(maxHeap) {
|
||||
// Инвертировать знак элемента
|
||||
return maxHeap.getMaxHeap().map((num) => -num);
|
||||
}
|
||||
|
||||
/* Найти k наибольших элементов массива с помощью кучи */
|
||||
function topKHeap(nums, k) {
|
||||
// Инициализация минимальной кучи
|
||||
// Обратите внимание: мы инвертируем все элементы кучи, чтобы с помощью максимальной кучи имитировать минимальную
|
||||
const maxHeap = new MaxHeap([]);
|
||||
// Поместить первые k элементов массива в кучу
|
||||
for (let i = 0; i < k; i++) {
|
||||
pushMinHeap(maxHeap, nums[i]);
|
||||
}
|
||||
// Начиная с элемента k+1, поддерживать длину кучи равной k
|
||||
for (let i = k; i < nums.length; i++) {
|
||||
// Если текущий элемент больше элемента на вершине кучи, извлечь вершину кучи и добавить текущий элемент в кучу
|
||||
if (nums[i] > peekMinHeap(maxHeap)) {
|
||||
popMinHeap(maxHeap);
|
||||
pushMinHeap(maxHeap, nums[i]);
|
||||
}
|
||||
}
|
||||
// Вернуть элементы кучи
|
||||
return getMinHeap(maxHeap);
|
||||
}
|
||||
|
||||
/* Driver Code */
|
||||
const nums = [1, 7, 6, 3, 2];
|
||||
const k = 3;
|
||||
const res = topKHeap(nums, k);
|
||||
console.log(`Наибольшие ${k} элементов`, res);
|
||||
Reference in New Issue
Block a user