mirror of
https://github.com/Xevion/Rebirth.git
synced 2025-12-15 00:12:48 -06:00
Bobbing, KDTree implementation
This commit is contained in:
211
Assets/Datastructures/Heap/BaseHeap.cs
Normal file
211
Assets/Datastructures/Heap/BaseHeap.cs
Normal file
@@ -0,0 +1,211 @@
|
||||
/*MIT License
|
||||
|
||||
Copyright(c) 2018 Vili Volčini / viliwonka
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
of this software and associated documentation files (the "Software"), to deal
|
||||
in the Software without restriction, including without limitation the rights
|
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
copies of the Software, and to permit persons to whom the Software is
|
||||
furnished to do so, subject to the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice shall be included in all
|
||||
copies or substantial portions of the Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
SOFTWARE.
|
||||
*/
|
||||
|
||||
using System.Collections.Generic;
|
||||
|
||||
namespace DataStructures.ViliWonka.Heap {
|
||||
|
||||
// array start at index 1, optimisation reason
|
||||
public abstract class BaseHeap {
|
||||
|
||||
protected int nodesCount;
|
||||
protected int maxSize;
|
||||
|
||||
protected float[] heap;
|
||||
|
||||
protected BaseHeap(int initialSize) {
|
||||
|
||||
maxSize = initialSize;
|
||||
heap = new float[initialSize + 1];
|
||||
}
|
||||
|
||||
public int Count { get { return nodesCount; } }
|
||||
|
||||
public float HeadValue { get { return heap[1]; } }
|
||||
|
||||
public void Clear() {
|
||||
nodesCount = 0;
|
||||
}
|
||||
|
||||
protected int Parent(int index) { return (index >> 1); }
|
||||
protected int Left (int index) { return (index << 1); }
|
||||
protected int Right (int index) { return (index << 1) | 1; }
|
||||
|
||||
// bubble down, MaxHeap version
|
||||
protected void BubbleDownMax(int index) {
|
||||
|
||||
int L = Left(index);
|
||||
int R = Right(index);
|
||||
|
||||
// bubbling down, 2 kids
|
||||
while (R <= nodesCount) {
|
||||
|
||||
// if heap property is violated between index and Left child
|
||||
if(heap[index] < heap[L]) {
|
||||
|
||||
if (heap[L] < heap[R]) {
|
||||
|
||||
Swap(index, R); // left has bigger priority
|
||||
index = R;
|
||||
}
|
||||
else {
|
||||
|
||||
Swap(index, L); // right has bigger priority
|
||||
index = L;
|
||||
}
|
||||
}
|
||||
else {
|
||||
// if heap property is violated between index and R
|
||||
if (heap[index] < heap[R]) {
|
||||
|
||||
Swap(index, R);
|
||||
index = R;
|
||||
}
|
||||
else {
|
||||
|
||||
index = L;
|
||||
L = Left(index);
|
||||
break;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
L = Left(index);
|
||||
R = Right(index);
|
||||
}
|
||||
|
||||
// only left & last children available to test and swap
|
||||
if (L <= nodesCount && heap[index] < heap[L]) {
|
||||
Swap(index, L);
|
||||
}
|
||||
}
|
||||
|
||||
// bubble up, MaxHeap version
|
||||
protected void BubbleUpMax(int index) {
|
||||
|
||||
int P = Parent(index);
|
||||
|
||||
//swap, until Heap property isn't violated anymore
|
||||
while (P > 0 && heap[P] < heap[index]) {
|
||||
|
||||
Swap(P, index);
|
||||
|
||||
index = P;
|
||||
P = Parent(index);
|
||||
}
|
||||
}
|
||||
|
||||
// bubble down, MinHeap version
|
||||
protected void BubbleDownMin(int index) {
|
||||
|
||||
int L = Left(index);
|
||||
int R = Right(index);
|
||||
|
||||
// bubbling down, 2 kids
|
||||
while(R <= nodesCount) {
|
||||
|
||||
// if heap property is violated between index and Left child
|
||||
if(heap[index] > heap[L]) {
|
||||
|
||||
if(heap[L] > heap[R]) {
|
||||
|
||||
Swap(index, R); // right has smaller priority
|
||||
index = R;
|
||||
}
|
||||
else {
|
||||
|
||||
Swap(index, L); // left has smaller priority
|
||||
index = L;
|
||||
}
|
||||
}
|
||||
else {
|
||||
// if heap property is violated between index and R
|
||||
if(heap[index] > heap[R]) {
|
||||
|
||||
Swap(index, R);
|
||||
index = R;
|
||||
}
|
||||
else {
|
||||
|
||||
index = L;
|
||||
L = Left(index);
|
||||
break;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
L = Left(index);
|
||||
R = Right(index);
|
||||
}
|
||||
|
||||
// only left & last children available to test and swap
|
||||
if(L <= nodesCount && heap[index] > heap[L]) {
|
||||
Swap(index, L);
|
||||
}
|
||||
}
|
||||
|
||||
// bubble up, MinHeap version
|
||||
protected void BubbleUpMin(int index) {
|
||||
|
||||
int P = Parent(index);
|
||||
|
||||
//swap, until Heap property isn't violated anymore
|
||||
while(P > 0 && heap[P] > heap[index]) {
|
||||
|
||||
Swap(P, index);
|
||||
|
||||
index = P;
|
||||
P = Parent(index);
|
||||
}
|
||||
}
|
||||
|
||||
protected float tempHeap;
|
||||
protected virtual void Swap(int A, int B) {
|
||||
|
||||
tempHeap = heap[A];
|
||||
heap[A] = heap[B];
|
||||
heap[B] = tempHeap;
|
||||
}
|
||||
|
||||
protected virtual void UpsizeHeap() {
|
||||
|
||||
maxSize *= 2;
|
||||
System.Array.Resize(ref heap, maxSize + 1);
|
||||
}
|
||||
|
||||
public virtual void PushValue(float h) {
|
||||
throw new System.NotImplementedException();
|
||||
}
|
||||
|
||||
public virtual float PopValue() {
|
||||
throw new System.NotImplementedException();
|
||||
}
|
||||
|
||||
public void FlushHeapResult(List<float> heapList) {
|
||||
|
||||
for(int i = 1; i < Count; i++) {
|
||||
heapList.Add(heap[i]);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
11
Assets/Datastructures/Heap/BaseHeap.cs.meta
Normal file
11
Assets/Datastructures/Heap/BaseHeap.cs.meta
Normal file
@@ -0,0 +1,11 @@
|
||||
fileFormatVersion: 2
|
||||
guid: f79974e0b20985b47a168ebcb1f3e74b
|
||||
MonoImporter:
|
||||
externalObjects: {}
|
||||
serializedVersion: 2
|
||||
defaultReferences: []
|
||||
executionOrder: 0
|
||||
icon: {instanceID: 0}
|
||||
userData:
|
||||
assetBundleName:
|
||||
assetBundleVariant:
|
||||
229
Assets/Datastructures/Heap/KSmallest.cs
Normal file
229
Assets/Datastructures/Heap/KSmallest.cs
Normal file
@@ -0,0 +1,229 @@
|
||||
/*MIT License
|
||||
|
||||
Copyright(c) 2018 Vili Volčini / viliwonka
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
of this software and associated documentation files (the "Software"), to deal
|
||||
in the Software without restriction, including without limitation the rights
|
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
copies of the Software, and to permit persons to whom the Software is
|
||||
furnished to do so, subject to the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice shall be included in all
|
||||
copies or substantial portions of the Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
SOFTWARE.
|
||||
*/
|
||||
|
||||
using System.Collections.Generic;
|
||||
|
||||
namespace DataStructures.ViliWonka.Heap {
|
||||
|
||||
public class KSmallestHeap : BaseHeap {
|
||||
|
||||
public KSmallestHeap(int maxEntries) : base(maxEntries) {
|
||||
|
||||
}
|
||||
|
||||
public bool Full {
|
||||
get {
|
||||
return maxSize == nodesCount;
|
||||
}
|
||||
}
|
||||
|
||||
// in lots of cases, max head gets removed
|
||||
public override void PushValue(float h) {
|
||||
|
||||
// if heap full
|
||||
if(nodesCount == maxSize) {
|
||||
|
||||
// if Heads priority is smaller than input priority, then ignore that item
|
||||
if(HeadValue < h) {
|
||||
|
||||
return;
|
||||
}
|
||||
else {
|
||||
|
||||
heap[1] = h; // remove top element
|
||||
BubbleDownMax(1); // bubble it down
|
||||
}
|
||||
}
|
||||
else {
|
||||
|
||||
nodesCount++;
|
||||
heap[nodesCount] = h;
|
||||
BubbleUpMax(nodesCount);
|
||||
}
|
||||
}
|
||||
|
||||
public override float PopValue() {
|
||||
|
||||
if(nodesCount == 0)
|
||||
throw new System.ArgumentException("Heap is empty!");
|
||||
|
||||
float result = heap[1];
|
||||
|
||||
heap[1] = heap[nodesCount];
|
||||
nodesCount--;
|
||||
BubbleDownMax(1);
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
public void Print() {
|
||||
|
||||
UnityEngine.Debug.Log("HeapPropertyHolds? " + HeapPropertyHolds(1));
|
||||
}
|
||||
|
||||
//should remove
|
||||
public bool HeapPropertyHolds(int index, int depth = 0) {
|
||||
|
||||
if (index > nodesCount)
|
||||
return true;
|
||||
|
||||
UnityEngine.Debug.Log(heap[index]);
|
||||
|
||||
int L = Left(index);
|
||||
int R = Right(index);
|
||||
|
||||
bool bothHold = true;
|
||||
|
||||
if(L <= nodesCount) {
|
||||
|
||||
UnityEngine.Debug.Log(heap[index] + " => " + heap[L]);
|
||||
|
||||
if (heap[index] < heap[L])
|
||||
bothHold = false;
|
||||
}
|
||||
|
||||
// if L <= nodesCount, then R <= nodesCount can also happen
|
||||
if (R <= nodesCount) {
|
||||
|
||||
UnityEngine.Debug.Log(heap[index] + " => " + heap[R]);
|
||||
|
||||
if (bothHold && heap[index] < heap[R])
|
||||
bothHold = false;
|
||||
|
||||
}
|
||||
|
||||
return bothHold & HeapPropertyHolds(L, depth + 1) & HeapPropertyHolds(R, depth + 1);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
// array start at index 1
|
||||
// generic version
|
||||
public class KSmallestHeap<T> : KSmallestHeap {
|
||||
|
||||
T[] objs; //objects
|
||||
|
||||
public KSmallestHeap(int maxEntries) : base(maxEntries) {
|
||||
objs = new T[maxEntries + 1];
|
||||
}
|
||||
|
||||
public T HeadHeapObject { get { return objs[1]; } }
|
||||
|
||||
T tempObjs;
|
||||
protected override void Swap(int A, int B) {
|
||||
|
||||
tempHeap = heap[A];
|
||||
tempObjs = objs[A];
|
||||
|
||||
heap[A] = heap[B];
|
||||
objs[A] = objs[B];
|
||||
|
||||
heap[B] = tempHeap;
|
||||
objs[B] = tempObjs;
|
||||
}
|
||||
|
||||
public override void PushValue(float h) {
|
||||
throw new System.ArgumentException("Use Push(T, float)!");
|
||||
}
|
||||
|
||||
public void PushObj(T obj, float h) {
|
||||
|
||||
// if heap full
|
||||
if(nodesCount == maxSize) {
|
||||
|
||||
// if Heads priority is smaller than input priority, then ignore that item
|
||||
if(HeadValue < h) {
|
||||
|
||||
return;
|
||||
}
|
||||
else {
|
||||
|
||||
heap[1] = h; // remove top element
|
||||
objs[1] = obj;
|
||||
BubbleDownMax(1); // bubble it down
|
||||
}
|
||||
}
|
||||
else {
|
||||
|
||||
nodesCount++;
|
||||
heap[nodesCount] = h;
|
||||
objs[nodesCount] = obj;
|
||||
BubbleUpMax(nodesCount);
|
||||
}
|
||||
}
|
||||
|
||||
public override float PopValue() {
|
||||
throw new System.ArgumentException("Use PopObj()!");
|
||||
}
|
||||
|
||||
public T PopObj() {
|
||||
|
||||
if(nodesCount == 0)
|
||||
throw new System.ArgumentException("Heap is empty!");
|
||||
|
||||
T result = objs[1];
|
||||
|
||||
heap[1] = heap[nodesCount];
|
||||
objs[1] = objs[nodesCount];
|
||||
|
||||
nodesCount--;
|
||||
BubbleDownMax(1);
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
public T PopObj(ref float heapValue) {
|
||||
|
||||
if(nodesCount == 0)
|
||||
throw new System.ArgumentException("Heap is empty!");
|
||||
|
||||
heapValue = heap[1];
|
||||
T result = PopObj();
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
//flush internal results, returns ordered data
|
||||
public void FlushResult(List<T> resultList, List<float> heapList = null) {
|
||||
|
||||
int count = nodesCount + 1;
|
||||
|
||||
|
||||
if(heapList == null) {
|
||||
|
||||
for(int i = 1; i < count; i++) {
|
||||
resultList.Add(PopObj());
|
||||
}
|
||||
}
|
||||
else {
|
||||
|
||||
float h = 0f;
|
||||
|
||||
for(int i = 1; i < count; i++) {
|
||||
resultList.Add(PopObj(ref h));
|
||||
heapList.Add(h);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
11
Assets/Datastructures/Heap/KSmallest.cs.meta
Normal file
11
Assets/Datastructures/Heap/KSmallest.cs.meta
Normal file
@@ -0,0 +1,11 @@
|
||||
fileFormatVersion: 2
|
||||
guid: 616a0666a4955ed4bbe7fcac27b32ae7
|
||||
MonoImporter:
|
||||
externalObjects: {}
|
||||
serializedVersion: 2
|
||||
defaultReferences: []
|
||||
executionOrder: 0
|
||||
icon: {instanceID: 0}
|
||||
userData:
|
||||
assetBundleName:
|
||||
assetBundleVariant:
|
||||
168
Assets/Datastructures/Heap/MaxHeap.cs
Normal file
168
Assets/Datastructures/Heap/MaxHeap.cs
Normal file
@@ -0,0 +1,168 @@
|
||||
/*MIT License
|
||||
|
||||
Copyright(c) 2018 Vili Volčini / viliwonka
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
of this software and associated documentation files (the "Software"), to deal
|
||||
in the Software without restriction, including without limitation the rights
|
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
copies of the Software, and to permit persons to whom the Software is
|
||||
furnished to do so, subject to the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice shall be included in all
|
||||
copies or substantial portions of the Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
SOFTWARE.
|
||||
*/
|
||||
|
||||
using System.Collections.Generic;
|
||||
|
||||
namespace DataStructures.ViliWonka.Heap {
|
||||
|
||||
public class MaxHeap : BaseHeap {
|
||||
|
||||
public MaxHeap(int initialSize = 2048) : base(initialSize) {
|
||||
|
||||
|
||||
}
|
||||
|
||||
public override void PushValue(float h) {
|
||||
|
||||
// if heap array is full
|
||||
if(nodesCount == maxSize) {
|
||||
|
||||
UpsizeHeap();
|
||||
}
|
||||
|
||||
nodesCount++;
|
||||
heap[nodesCount] = h;
|
||||
BubbleUpMax(nodesCount);
|
||||
}
|
||||
|
||||
public override float PopValue() {
|
||||
|
||||
if(nodesCount == 0)
|
||||
throw new System.ArgumentException("Heap is empty!");
|
||||
|
||||
float result = heap[1];
|
||||
|
||||
heap[1] = heap[nodesCount];
|
||||
nodesCount--;
|
||||
BubbleDownMax(1);
|
||||
|
||||
return result;
|
||||
}
|
||||
}
|
||||
|
||||
// generic version
|
||||
public class MaxHeap<T> : MaxHeap {
|
||||
|
||||
T[] objs; // objects
|
||||
|
||||
public MaxHeap(int maxNodes) : base(maxNodes) {
|
||||
objs = new T[maxNodes + 1];
|
||||
}
|
||||
|
||||
public T HeadHeapObject { get { return objs[1]; } }
|
||||
|
||||
T tempObjs;
|
||||
protected override void Swap(int A, int B) {
|
||||
|
||||
tempHeap = heap[A];
|
||||
tempObjs = objs[A];
|
||||
|
||||
heap[A] = heap[B];
|
||||
objs[A] = objs[B];
|
||||
|
||||
heap[B] = tempHeap;
|
||||
objs[B] = tempObjs;
|
||||
}
|
||||
|
||||
public override void PushValue(float h) {
|
||||
throw new System.ArgumentException("Use PushObj(T, float)!");
|
||||
}
|
||||
|
||||
public override float PopValue() {
|
||||
throw new System.ArgumentException("Use Push(T, float)!");
|
||||
}
|
||||
|
||||
public void PushObj(T obj, float h) {
|
||||
|
||||
// if heap array is full
|
||||
if(nodesCount == maxSize) {
|
||||
UpsizeHeap();
|
||||
}
|
||||
|
||||
nodesCount++;
|
||||
heap[nodesCount] = h;
|
||||
objs[nodesCount] = obj;
|
||||
|
||||
BubbleUpMin(nodesCount);
|
||||
}
|
||||
|
||||
public T PopObj() {
|
||||
|
||||
if(nodesCount == 0)
|
||||
throw new System.ArgumentException("Heap is empty!");
|
||||
|
||||
T result = objs[1];
|
||||
|
||||
heap[1] = heap[nodesCount];
|
||||
objs[1] = objs[nodesCount];
|
||||
|
||||
objs[nodesCount] = default(T);
|
||||
|
||||
nodesCount--;
|
||||
BubbleDownMin(1);
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
|
||||
public T PopObj(ref float heapValue) {
|
||||
|
||||
if(nodesCount == 0)
|
||||
throw new System.ArgumentException("Heap is empty!");
|
||||
|
||||
heapValue = heap[1];
|
||||
T result = PopObj();
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
protected override void UpsizeHeap() {
|
||||
|
||||
maxSize *= 2;
|
||||
System.Array.Resize(ref heap, maxSize + 1);
|
||||
System.Array.Resize(ref objs, maxSize + 1);
|
||||
}
|
||||
|
||||
//flush internal results, returns ordered data
|
||||
public void FlushResult(List<T> resultList, List<float> heapList = null) {
|
||||
|
||||
int count = nodesCount + 1;
|
||||
|
||||
if(heapList == null) {
|
||||
|
||||
for(int i = 1; i < count; i++) {
|
||||
resultList.Add(PopObj());
|
||||
}
|
||||
}
|
||||
else {
|
||||
|
||||
float h = 0f;
|
||||
|
||||
for(int i = 1; i < count; i++) {
|
||||
resultList.Add(PopObj(ref h));
|
||||
heapList.Add(h);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
11
Assets/Datastructures/Heap/MaxHeap.cs.meta
Normal file
11
Assets/Datastructures/Heap/MaxHeap.cs.meta
Normal file
@@ -0,0 +1,11 @@
|
||||
fileFormatVersion: 2
|
||||
guid: 6c07f133f8057584fadbaf01edc5fd3a
|
||||
MonoImporter:
|
||||
externalObjects: {}
|
||||
serializedVersion: 2
|
||||
defaultReferences: []
|
||||
executionOrder: 0
|
||||
icon: {instanceID: 0}
|
||||
userData:
|
||||
assetBundleName:
|
||||
assetBundleVariant:
|
||||
171
Assets/Datastructures/Heap/MinHeap.cs
Normal file
171
Assets/Datastructures/Heap/MinHeap.cs
Normal file
@@ -0,0 +1,171 @@
|
||||
/*MIT License
|
||||
|
||||
Copyright(c) 2018 Vili Volčini / viliwonka
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
of this software and associated documentation files (the "Software"), to deal
|
||||
in the Software without restriction, including without limitation the rights
|
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
copies of the Software, and to permit persons to whom the Software is
|
||||
furnished to do so, subject to the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice shall be included in all
|
||||
copies or substantial portions of the Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
SOFTWARE.
|
||||
*/
|
||||
|
||||
using System.Collections.Generic;
|
||||
|
||||
namespace DataStructures.ViliWonka.Heap {
|
||||
|
||||
public class MinHeap : BaseHeap {
|
||||
|
||||
public MinHeap(int initialSize = 2048) : base(initialSize) {
|
||||
|
||||
}
|
||||
|
||||
public override void PushValue(float h) {
|
||||
|
||||
// if heap array is full
|
||||
if(nodesCount == maxSize) {
|
||||
|
||||
UpsizeHeap();
|
||||
}
|
||||
|
||||
nodesCount++;
|
||||
heap[nodesCount] = h;
|
||||
BubbleUpMin(nodesCount);
|
||||
}
|
||||
|
||||
public override float PopValue() {
|
||||
|
||||
if(nodesCount == 0)
|
||||
throw new System.ArgumentException("Heap is empty!");
|
||||
|
||||
float result = heap[1];
|
||||
|
||||
heap[1] = heap[nodesCount];
|
||||
|
||||
nodesCount--;
|
||||
|
||||
if(nodesCount != 0)
|
||||
BubbleDownMin(1);
|
||||
|
||||
return result;
|
||||
}
|
||||
}
|
||||
|
||||
// generic version
|
||||
public class MinHeap<T> : MinHeap {
|
||||
|
||||
T[] objs; // objects
|
||||
|
||||
public MinHeap(int maxNodes = 2048) : base(maxNodes) {
|
||||
objs = new T[maxNodes + 1];
|
||||
}
|
||||
|
||||
public T HeadHeapObject { get { return objs[1]; } }
|
||||
|
||||
T tempObjs;
|
||||
protected override void Swap(int A, int B) {
|
||||
|
||||
tempHeap = heap[A];
|
||||
tempObjs = objs[A];
|
||||
|
||||
heap[A] = heap[B];
|
||||
objs[A] = objs[B];
|
||||
|
||||
heap[B] = tempHeap;
|
||||
objs[B] = tempObjs;
|
||||
}
|
||||
|
||||
public override void PushValue(float h) {
|
||||
throw new System.ArgumentException("Use Push(T, float)!");
|
||||
}
|
||||
|
||||
public override float PopValue() {
|
||||
throw new System.ArgumentException("Use Push(T, float)!");
|
||||
}
|
||||
|
||||
public void PushObj(T obj, float h) {
|
||||
|
||||
// if heap array is full
|
||||
if(nodesCount == maxSize) {
|
||||
UpsizeHeap();
|
||||
}
|
||||
|
||||
nodesCount++;
|
||||
heap[nodesCount] = h;
|
||||
objs[nodesCount] = obj;
|
||||
|
||||
BubbleUpMin(nodesCount);
|
||||
}
|
||||
|
||||
public T PopObj() {
|
||||
|
||||
if(nodesCount == 0)
|
||||
throw new System.ArgumentException("Heap is empty!");
|
||||
|
||||
T result = objs[1];
|
||||
|
||||
heap[1] = heap[nodesCount];
|
||||
objs[1] = objs[nodesCount];
|
||||
|
||||
objs[nodesCount] = default(T);
|
||||
|
||||
nodesCount--;
|
||||
|
||||
if(nodesCount != 0)
|
||||
BubbleDownMin(1);
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
public T PopObj(ref float heapValue) {
|
||||
|
||||
if(nodesCount == 0)
|
||||
throw new System.ArgumentException("Heap is empty!");
|
||||
|
||||
heapValue = heap[1];
|
||||
T result = PopObj();
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
protected override void UpsizeHeap() {
|
||||
|
||||
maxSize *= 2;
|
||||
System.Array.Resize(ref heap, maxSize + 1);
|
||||
System.Array.Resize(ref objs, maxSize + 1);
|
||||
}
|
||||
|
||||
//flush internal array, returns ordered data
|
||||
public void FlushResult(List<T> resultList, List<float> heapList = null) {
|
||||
|
||||
int count = nodesCount + 1;
|
||||
|
||||
if(heapList == null) {
|
||||
|
||||
for(int i = 1; i < count; i++) {
|
||||
resultList.Add(PopObj());
|
||||
}
|
||||
}
|
||||
else {
|
||||
|
||||
float h = 0f;
|
||||
|
||||
for(int i = 1; i < count; i++) {
|
||||
resultList.Add(PopObj(ref h));
|
||||
heapList.Add(h);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
11
Assets/Datastructures/Heap/MinHeap.cs.meta
Normal file
11
Assets/Datastructures/Heap/MinHeap.cs.meta
Normal file
@@ -0,0 +1,11 @@
|
||||
fileFormatVersion: 2
|
||||
guid: 354799470fae8d74d8f977dd8db040cf
|
||||
MonoImporter:
|
||||
externalObjects: {}
|
||||
serializedVersion: 2
|
||||
defaultReferences: []
|
||||
executionOrder: 0
|
||||
icon: {instanceID: 0}
|
||||
userData:
|
||||
assetBundleName:
|
||||
assetBundleVariant:
|
||||
298
Assets/Datastructures/Heap/heapTestScene.unity
Normal file
298
Assets/Datastructures/Heap/heapTestScene.unity
Normal file
@@ -0,0 +1,298 @@
|
||||
%YAML 1.1
|
||||
%TAG !u! tag:unity3d.com,2011:
|
||||
--- !u!29 &1
|
||||
OcclusionCullingSettings:
|
||||
m_ObjectHideFlags: 0
|
||||
serializedVersion: 2
|
||||
m_OcclusionBakeSettings:
|
||||
smallestOccluder: 5
|
||||
smallestHole: 0.25
|
||||
backfaceThreshold: 100
|
||||
m_SceneGUID: 00000000000000000000000000000000
|
||||
m_OcclusionCullingData: {fileID: 0}
|
||||
--- !u!104 &2
|
||||
RenderSettings:
|
||||
m_ObjectHideFlags: 0
|
||||
serializedVersion: 9
|
||||
m_Fog: 0
|
||||
m_FogColor: {r: 0.5, g: 0.5, b: 0.5, a: 1}
|
||||
m_FogMode: 3
|
||||
m_FogDensity: 0.01
|
||||
m_LinearFogStart: 0
|
||||
m_LinearFogEnd: 300
|
||||
m_AmbientSkyColor: {r: 0.212, g: 0.227, b: 0.259, a: 1}
|
||||
m_AmbientEquatorColor: {r: 0.114, g: 0.125, b: 0.133, a: 1}
|
||||
m_AmbientGroundColor: {r: 0.047, g: 0.043, b: 0.035, a: 1}
|
||||
m_AmbientIntensity: 1
|
||||
m_AmbientMode: 0
|
||||
m_SubtractiveShadowColor: {r: 0.42, g: 0.478, b: 0.627, a: 1}
|
||||
m_SkyboxMaterial: {fileID: 10304, guid: 0000000000000000f000000000000000, type: 0}
|
||||
m_HaloStrength: 0.5
|
||||
m_FlareStrength: 1
|
||||
m_FlareFadeSpeed: 3
|
||||
m_HaloTexture: {fileID: 0}
|
||||
m_SpotCookie: {fileID: 10001, guid: 0000000000000000e000000000000000, type: 0}
|
||||
m_DefaultReflectionMode: 0
|
||||
m_DefaultReflectionResolution: 128
|
||||
m_ReflectionBounces: 1
|
||||
m_ReflectionIntensity: 1
|
||||
m_CustomReflection: {fileID: 0}
|
||||
m_Sun: {fileID: 0}
|
||||
m_IndirectSpecularColor: {r: 0.44657898, g: 0.4964133, b: 0.5748178, a: 1}
|
||||
m_UseRadianceAmbientProbe: 0
|
||||
--- !u!157 &3
|
||||
LightmapSettings:
|
||||
m_ObjectHideFlags: 0
|
||||
serializedVersion: 11
|
||||
m_GIWorkflowMode: 0
|
||||
m_GISettings:
|
||||
serializedVersion: 2
|
||||
m_BounceScale: 1
|
||||
m_IndirectOutputScale: 1
|
||||
m_AlbedoBoost: 1
|
||||
m_TemporalCoherenceThreshold: 1
|
||||
m_EnvironmentLightingMode: 0
|
||||
m_EnableBakedLightmaps: 1
|
||||
m_EnableRealtimeLightmaps: 1
|
||||
m_LightmapEditorSettings:
|
||||
serializedVersion: 10
|
||||
m_Resolution: 2
|
||||
m_BakeResolution: 40
|
||||
m_AtlasSize: 1024
|
||||
m_AO: 0
|
||||
m_AOMaxDistance: 1
|
||||
m_CompAOExponent: 1
|
||||
m_CompAOExponentDirect: 0
|
||||
m_Padding: 2
|
||||
m_LightmapParameters: {fileID: 0}
|
||||
m_LightmapsBakeMode: 1
|
||||
m_TextureCompression: 1
|
||||
m_FinalGather: 0
|
||||
m_FinalGatherFiltering: 1
|
||||
m_FinalGatherRayCount: 256
|
||||
m_ReflectionCompression: 2
|
||||
m_MixedBakeMode: 2
|
||||
m_BakeBackend: 1
|
||||
m_PVRSampling: 1
|
||||
m_PVRDirectSampleCount: 32
|
||||
m_PVRSampleCount: 500
|
||||
m_PVRBounces: 2
|
||||
m_PVRFilterTypeDirect: 0
|
||||
m_PVRFilterTypeIndirect: 0
|
||||
m_PVRFilterTypeAO: 0
|
||||
m_PVRFilteringMode: 1
|
||||
m_PVRCulling: 1
|
||||
m_PVRFilteringGaussRadiusDirect: 1
|
||||
m_PVRFilteringGaussRadiusIndirect: 5
|
||||
m_PVRFilteringGaussRadiusAO: 2
|
||||
m_PVRFilteringAtrousPositionSigmaDirect: 0.5
|
||||
m_PVRFilteringAtrousPositionSigmaIndirect: 2
|
||||
m_PVRFilteringAtrousPositionSigmaAO: 1
|
||||
m_ShowResolutionOverlay: 1
|
||||
m_LightingDataAsset: {fileID: 0}
|
||||
m_UseShadowmask: 1
|
||||
--- !u!196 &4
|
||||
NavMeshSettings:
|
||||
serializedVersion: 2
|
||||
m_ObjectHideFlags: 0
|
||||
m_BuildSettings:
|
||||
serializedVersion: 2
|
||||
agentTypeID: 0
|
||||
agentRadius: 0.5
|
||||
agentHeight: 2
|
||||
agentSlope: 45
|
||||
agentClimb: 0.4
|
||||
ledgeDropHeight: 0
|
||||
maxJumpAcrossDistance: 0
|
||||
minRegionArea: 2
|
||||
manualCellSize: 0
|
||||
cellSize: 0.16666667
|
||||
manualTileSize: 0
|
||||
tileSize: 256
|
||||
accuratePlacement: 0
|
||||
debug:
|
||||
m_Flags: 0
|
||||
m_NavMeshData: {fileID: 0}
|
||||
--- !u!1 &653832527
|
||||
GameObject:
|
||||
m_ObjectHideFlags: 0
|
||||
m_CorrespondingSourceObject: {fileID: 0}
|
||||
m_PrefabInternal: {fileID: 0}
|
||||
serializedVersion: 6
|
||||
m_Component:
|
||||
- component: {fileID: 653832530}
|
||||
- component: {fileID: 653832529}
|
||||
- component: {fileID: 653832528}
|
||||
m_Layer: 0
|
||||
m_Name: Main Camera
|
||||
m_TagString: MainCamera
|
||||
m_Icon: {fileID: 0}
|
||||
m_NavMeshLayer: 0
|
||||
m_StaticEditorFlags: 0
|
||||
m_IsActive: 1
|
||||
--- !u!81 &653832528
|
||||
AudioListener:
|
||||
m_ObjectHideFlags: 0
|
||||
m_CorrespondingSourceObject: {fileID: 0}
|
||||
m_PrefabInternal: {fileID: 0}
|
||||
m_GameObject: {fileID: 653832527}
|
||||
m_Enabled: 1
|
||||
--- !u!20 &653832529
|
||||
Camera:
|
||||
m_ObjectHideFlags: 0
|
||||
m_CorrespondingSourceObject: {fileID: 0}
|
||||
m_PrefabInternal: {fileID: 0}
|
||||
m_GameObject: {fileID: 653832527}
|
||||
m_Enabled: 1
|
||||
serializedVersion: 2
|
||||
m_ClearFlags: 1
|
||||
m_BackGroundColor: {r: 0.19215687, g: 0.3019608, b: 0.4745098, a: 0}
|
||||
m_projectionMatrixMode: 1
|
||||
m_SensorSize: {x: 36, y: 24}
|
||||
m_LensShift: {x: 0, y: 0}
|
||||
m_FocalLength: 50
|
||||
m_NormalizedViewPortRect:
|
||||
serializedVersion: 2
|
||||
x: 0
|
||||
y: 0
|
||||
width: 1
|
||||
height: 1
|
||||
near clip plane: 0.3
|
||||
far clip plane: 1000
|
||||
field of view: 60
|
||||
orthographic: 0
|
||||
orthographic size: 5
|
||||
m_Depth: -1
|
||||
m_CullingMask:
|
||||
serializedVersion: 2
|
||||
m_Bits: 4294967295
|
||||
m_RenderingPath: -1
|
||||
m_TargetTexture: {fileID: 0}
|
||||
m_TargetDisplay: 0
|
||||
m_TargetEye: 3
|
||||
m_HDR: 1
|
||||
m_AllowMSAA: 1
|
||||
m_AllowDynamicResolution: 0
|
||||
m_ForceIntoRT: 0
|
||||
m_OcclusionCulling: 1
|
||||
m_StereoConvergence: 10
|
||||
m_StereoSeparation: 0.022
|
||||
--- !u!4 &653832530
|
||||
Transform:
|
||||
m_ObjectHideFlags: 0
|
||||
m_CorrespondingSourceObject: {fileID: 0}
|
||||
m_PrefabInternal: {fileID: 0}
|
||||
m_GameObject: {fileID: 653832527}
|
||||
m_LocalRotation: {x: 0, y: 0, z: 0, w: 1}
|
||||
m_LocalPosition: {x: 0, y: 1, z: -10}
|
||||
m_LocalScale: {x: 1, y: 1, z: 1}
|
||||
m_Children: []
|
||||
m_Father: {fileID: 0}
|
||||
m_RootOrder: 0
|
||||
m_LocalEulerAnglesHint: {x: 0, y: 0, z: 0}
|
||||
--- !u!1 &1730864583
|
||||
GameObject:
|
||||
m_ObjectHideFlags: 0
|
||||
m_CorrespondingSourceObject: {fileID: 0}
|
||||
m_PrefabInternal: {fileID: 0}
|
||||
serializedVersion: 6
|
||||
m_Component:
|
||||
- component: {fileID: 1730864585}
|
||||
- component: {fileID: 1730864584}
|
||||
m_Layer: 0
|
||||
m_Name: Directional Light
|
||||
m_TagString: Untagged
|
||||
m_Icon: {fileID: 0}
|
||||
m_NavMeshLayer: 0
|
||||
m_StaticEditorFlags: 0
|
||||
m_IsActive: 1
|
||||
--- !u!108 &1730864584
|
||||
Light:
|
||||
m_ObjectHideFlags: 0
|
||||
m_CorrespondingSourceObject: {fileID: 0}
|
||||
m_PrefabInternal: {fileID: 0}
|
||||
m_GameObject: {fileID: 1730864583}
|
||||
m_Enabled: 1
|
||||
serializedVersion: 8
|
||||
m_Type: 1
|
||||
m_Color: {r: 1, g: 0.95686275, b: 0.8392157, a: 1}
|
||||
m_Intensity: 1
|
||||
m_Range: 10
|
||||
m_SpotAngle: 30
|
||||
m_CookieSize: 10
|
||||
m_Shadows:
|
||||
m_Type: 2
|
||||
m_Resolution: -1
|
||||
m_CustomResolution: -1
|
||||
m_Strength: 1
|
||||
m_Bias: 0.05
|
||||
m_NormalBias: 0.4
|
||||
m_NearPlane: 0.2
|
||||
m_Cookie: {fileID: 0}
|
||||
m_DrawHalo: 0
|
||||
m_Flare: {fileID: 0}
|
||||
m_RenderMode: 0
|
||||
m_CullingMask:
|
||||
serializedVersion: 2
|
||||
m_Bits: 4294967295
|
||||
m_Lightmapping: 4
|
||||
m_LightShadowCasterMode: 0
|
||||
m_AreaSize: {x: 1, y: 1}
|
||||
m_BounceIntensity: 1
|
||||
m_ColorTemperature: 6570
|
||||
m_UseColorTemperature: 0
|
||||
m_ShadowRadius: 0
|
||||
m_ShadowAngle: 0
|
||||
--- !u!4 &1730864585
|
||||
Transform:
|
||||
m_ObjectHideFlags: 0
|
||||
m_CorrespondingSourceObject: {fileID: 0}
|
||||
m_PrefabInternal: {fileID: 0}
|
||||
m_GameObject: {fileID: 1730864583}
|
||||
m_LocalRotation: {x: 0.40821788, y: -0.23456968, z: 0.10938163, w: 0.8754261}
|
||||
m_LocalPosition: {x: 0, y: 3, z: 0}
|
||||
m_LocalScale: {x: 1, y: 1, z: 1}
|
||||
m_Children: []
|
||||
m_Father: {fileID: 0}
|
||||
m_RootOrder: 1
|
||||
m_LocalEulerAnglesHint: {x: 50, y: -30, z: 0}
|
||||
--- !u!1 &1883466098
|
||||
GameObject:
|
||||
m_ObjectHideFlags: 0
|
||||
m_CorrespondingSourceObject: {fileID: 0}
|
||||
m_PrefabInternal: {fileID: 0}
|
||||
serializedVersion: 6
|
||||
m_Component:
|
||||
- component: {fileID: 1883466100}
|
||||
- component: {fileID: 1883466099}
|
||||
m_Layer: 0
|
||||
m_Name: GameObject
|
||||
m_TagString: Untagged
|
||||
m_Icon: {fileID: 0}
|
||||
m_NavMeshLayer: 0
|
||||
m_StaticEditorFlags: 0
|
||||
m_IsActive: 1
|
||||
--- !u!114 &1883466099
|
||||
MonoBehaviour:
|
||||
m_ObjectHideFlags: 0
|
||||
m_CorrespondingSourceObject: {fileID: 0}
|
||||
m_PrefabInternal: {fileID: 0}
|
||||
m_GameObject: {fileID: 1883466098}
|
||||
m_Enabled: 1
|
||||
m_EditorHideFlags: 0
|
||||
m_Script: {fileID: 11500000, guid: e3f22e25c6d555d45b86d62ecaaf8895, type: 3}
|
||||
m_Name:
|
||||
m_EditorClassIdentifier:
|
||||
--- !u!4 &1883466100
|
||||
Transform:
|
||||
m_ObjectHideFlags: 0
|
||||
m_CorrespondingSourceObject: {fileID: 0}
|
||||
m_PrefabInternal: {fileID: 0}
|
||||
m_GameObject: {fileID: 1883466098}
|
||||
m_LocalRotation: {x: 0, y: 0, z: 0, w: 1}
|
||||
m_LocalPosition: {x: 1.6106011, y: 0.41596547, z: 1.608385}
|
||||
m_LocalScale: {x: 1, y: 1, z: 1}
|
||||
m_Children: []
|
||||
m_Father: {fileID: 0}
|
||||
m_RootOrder: 2
|
||||
m_LocalEulerAnglesHint: {x: 0, y: 0, z: 0}
|
||||
7
Assets/Datastructures/Heap/heapTestScene.unity.meta
Normal file
7
Assets/Datastructures/Heap/heapTestScene.unity.meta
Normal file
@@ -0,0 +1,7 @@
|
||||
fileFormatVersion: 2
|
||||
guid: 3bd0cd1e56a2ce749a627974769aee70
|
||||
DefaultImporter:
|
||||
externalObjects: {}
|
||||
userData:
|
||||
assetBundleName:
|
||||
assetBundleVariant:
|
||||
Reference in New Issue
Block a user