Merge pull request #16501 from eugenp/fix-java-stream-integers-median-using-heap

Change int to double
This commit is contained in:
Ulisses Lima 2024-04-26 15:24:27 -03:00 committed by GitHub
commit fe759e41e4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 4 deletions

View File

@ -29,13 +29,13 @@ public class MedianOfIntegerStream {
} }
double getMedian() { double getMedian() {
int median; double median;
if (minHeap.size() < maxHeap.size()) { if (minHeap.size() < maxHeap.size()) {
median = maxHeap.peek(); median = maxHeap.peek();
} else if (minHeap.size() > maxHeap.size()) { } else if (minHeap.size() > maxHeap.size()) {
median = minHeap.peek(); median = minHeap.peek();
} else { } else {
median = (minHeap.peek() + maxHeap.peek()) / 2; median = (minHeap.peek() + maxHeap.peek()) / 2.0;
} }
return median; return median;
} }

View File

@ -25,11 +25,11 @@ public class MedianOfIntegerStream2 {
} }
double getMedian() { double getMedian() {
int median; double median;
if (minHeap.size() > maxHeap.size()) { if (minHeap.size() > maxHeap.size()) {
median = minHeap.peek(); median = minHeap.peek();
} else { } else {
median = (minHeap.peek() + maxHeap.peek()) / 2; median = (minHeap.peek() + maxHeap.peek()) / 2.0;
} }
return median; return median;
} }