algorithm/Java
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Java.git synced 2025-07-06 09:06:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add index validation to Min Heap and Max Heap (#3189)

Browse Source 
Co-authored-by: Andrii Siriak <siryaka@gmail.com>
This commit is contained in:
Jainam Kothari
2022-07-13 23:10:21 +05:30
committed by GitHub
parent b2f6827c36
commit 1a9937c7cb
3 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
src/main/java/com/thealgorithms/datastructures/heaps/GenericHeap → src/main/java/com/thealgorithms/datastructures/heaps/GenericHeap.java
View File

4
src/main/java/com/thealgorithms/datastructures/heaps/MaxHeap.java
View File

@ -43,6 +43,10 @@ public class MaxHeap implements Heap {
// Get the key of the element at a given index
private double getElementKey(int elementIndex) {
if ((elementIndex <= 0) || (elementIndex > maxHeap.size())) {
throw new IndexOutOfBoundsException("Index out of heap range");
}
return maxHeap.get(elementIndex - 1).getKey();
}

4
src/main/java/com/thealgorithms/datastructures/heaps/MinHeap.java
View File

@ -37,6 +37,10 @@ public class MinHeap implements Heap {
// Get the key of the element at a given index
private double getElementKey(int elementIndex) {
if ((elementIndex <= 0) || (elementIndex > minHeap.size())) {
throw new IndexOutOfBoundsException("Index out of heap range");
}
return minHeap.get(elementIndex - 1).getKey();
}