mirror of
https://github.com/krahets/hello-algo.git
synced 2026-06-15 14:48:05 +08:00
65 lines
2.2 KiB
C#
65 lines
2.2 KiB
C#
/**
|
||
* File: heap.cs
|
||
* Created Time: 2023-02-06
|
||
* Author: zjkung1123 (zjkung1123@gmail.com)
|
||
*/
|
||
|
||
namespace hello_algo.chapter_heap;
|
||
|
||
public class heap {
|
||
void TestPush(PriorityQueue<int, int> heap, int val) {
|
||
heap.Enqueue(val, val); // 要素をヒープに追加
|
||
Console.WriteLine($"\n要素 {val} をヒープに追加した後\n");
|
||
PrintUtil.PrintHeap(heap);
|
||
}
|
||
|
||
void TestPop(PriorityQueue<int, int> heap) {
|
||
int val = heap.Dequeue(); // ヒープ頂点の要素を取り出す
|
||
Console.WriteLine($"\nヒープの先頭要素 {val} を取り出した後\n");
|
||
PrintUtil.PrintHeap(heap);
|
||
}
|
||
|
||
[Test]
|
||
public void Test() {
|
||
/* ヒープを初期化 */
|
||
// 最小ヒープを初期化
|
||
PriorityQueue<int, int> minHeap = new();
|
||
// 最大ヒープを初期化する（Comparer はラムダ式で変更できる）
|
||
PriorityQueue<int, int> maxHeap = new(Comparer<int>.Create((x, y) => y.CompareTo(x)));
|
||
Console.WriteLine("以下のテストケースは最大ヒープです");
|
||
|
||
/* 要素をヒープに追加 */
|
||
TestPush(maxHeap, 1);
|
||
TestPush(maxHeap, 3);
|
||
TestPush(maxHeap, 2);
|
||
TestPush(maxHeap, 5);
|
||
TestPush(maxHeap, 4);
|
||
|
||
/* ヒープ頂点の要素を取得 */
|
||
int peek = maxHeap.Peek();
|
||
Console.WriteLine($"ヒープの先頭要素は {peek}");
|
||
|
||
/* ヒープ頂点の要素を取り出す */
|
||
// ヒープから取り出した要素は大きい順に並ぶ
|
||
TestPop(maxHeap);
|
||
TestPop(maxHeap);
|
||
TestPop(maxHeap);
|
||
TestPop(maxHeap);
|
||
TestPop(maxHeap);
|
||
|
||
/* ヒープのサイズを取得 */
|
||
int size = maxHeap.Count;
|
||
Console.WriteLine($"ヒープの要素数は {size}");
|
||
|
||
/* ヒープが空かどうかを判定 */
|
||
bool isEmpty = maxHeap.Count == 0;
|
||
Console.WriteLine($"ヒープは空か {isEmpty}");
|
||
|
||
/* リストを入力してヒープを構築 */
|
||
var list = new int[] { 1, 3, 2, 5, 4 };
|
||
minHeap = new PriorityQueue<int, int>(list.Select(x => (x, x)));
|
||
Console.WriteLine("リストを入力して最小ヒープを構築した後");
|
||
PrintUtil.PrintHeap(minHeap);
|
||
}
|
||
}
|