Files
2025-08-13 09:26:42 +08:00

161 lines
2.8 KiB
C#

using System;
using System.Collections.Generic;
using System.Threading;
namespace UniRx.InternalUtil
{
internal class PriorityQueue<T> where T : IComparable<T>
{
private struct IndexedItem : IComparable<IndexedItem>
{
public T Value;
public long Id;
public int CompareTo(IndexedItem other)
{
int num = Value.CompareTo(other.Value);
if (num == 0)
{
num = Id.CompareTo(other.Id);
}
return num;
}
}
private static long _count = long.MinValue;
private IndexedItem[] _items;
private int _size;
public int Count
{
get
{
return _size;
}
}
public PriorityQueue()
: this(16)
{
}
public PriorityQueue(int capacity)
{
_items = new IndexedItem[capacity];
_size = 0;
}
private bool IsHigherPriority(int left, int right)
{
return _items[left].CompareTo(_items[right]) < 0;
}
private void Percolate(int index)
{
if (index < _size && index >= 0)
{
int num = (index - 1) / 2;
if (num >= 0 && num != index && IsHigherPriority(index, num))
{
IndexedItem indexedItem = _items[index];
_items[index] = _items[num];
_items[num] = indexedItem;
Percolate(num);
}
}
}
private void Heapify()
{
Heapify(0);
}
private void Heapify(int index)
{
if (index < _size && index >= 0)
{
int num = 2 * index + 1;
int num2 = 2 * index + 2;
int num3 = index;
if (num < _size && IsHigherPriority(num, num3))
{
num3 = num;
}
if (num2 < _size && IsHigherPriority(num2, num3))
{
num3 = num2;
}
if (num3 != index)
{
IndexedItem indexedItem = _items[index];
_items[index] = _items[num3];
_items[num3] = indexedItem;
Heapify(num3);
}
}
}
public T Peek()
{
if (_size == 0)
{
throw new InvalidOperationException("HEAP is Empty");
}
return _items[0].Value;
}
private void RemoveAt(int index)
{
_items[index] = _items[--_size];
_items[_size] = default(IndexedItem);
Heapify();
if (_size < _items.Length / 4)
{
IndexedItem[] items = _items;
_items = new IndexedItem[_items.Length / 2];
Array.Copy(items, 0, _items, 0, _size);
}
}
public T Dequeue()
{
T result = Peek();
RemoveAt(0);
return result;
}
public void Enqueue(T item)
{
if (_size >= _items.Length)
{
IndexedItem[] items = _items;
_items = new IndexedItem[_items.Length * 2];
Array.Copy(items, _items, items.Length);
}
int num = _size++;
_items[num] = new IndexedItem
{
Value = item,
Id = Interlocked.Increment(ref _count)
};
Percolate(num);
}
public bool Remove(T item)
{
for (int i = 0; i < _size; i++)
{
if (EqualityComparer<T>.Default.Equals(_items[i].Value, item))
{
RemoveAt(i);
return true;
}
}
return false;
}
}
}