fix: NumericHistogram handle NaN cases (prestodb#25793)

## Description
fix: NumericHistogram handle NaN cases
Previous it will fail when value is Nan, and the penalty becomes Nan,
which is larger than Inf. So it breaks the assumptions: right is
guaranteed to exist because we set the penalty of the last bucket to
infinity.

## Motivation and Context
The unit test will fail before the fix and succeed after fix.

## Impact
fix some edge cases for the function

## Test Plan
Unit Tests

## Contributor checklist

- [x] Please make sure your submission complies with our [contributing
guide](https://github.com/prestodb/presto/blob/master/CONTRIBUTING.md),
in particular [code
style](https://github.com/prestodb/presto/blob/master/CONTRIBUTING.md#code-style)
and [commit
standards](https://github.com/prestodb/presto/blob/master/CONTRIBUTING.md#commit-standards).
- [x] PR description addresses the issue accurately and concisely. If
the change is non-trivial, a GitHub Issue is referenced.
- [x] Documented new properties (with its default value), SQL syntax,
functions, or other functionality.
- [x] If release notes are required, they follow the [release notes
guidelines](https://github.com/prestodb/presto/wiki/Release-Notes-Guidelines).
- [x] Adequate tests were added if applicable.
- [x] CI passed.

## Release Notes

```
== NO RELEASE NOTE ==
```

Author: skyelves

presto-main-base/src/main/java/com/facebook/presto/operator/aggregation/NumericHistogram.java

@@ -184,7 +184,7 @@ private static PriorityQueue<Entry> mergeBuckets(double[] values, double[] weigh
 
             Entry right = current.getRight();
 
-            // right is guaranteed to exist because we set the penalty of the last bucket to infinity
+            // right is guaranteed to exist because we set the penalty of the last bucket to NaN
             // so the first current in the queue can never be the last bucket
             checkState(right != null, "Expected right to be != null");
             checkState(right.isValid(), "Expected right to be valid");
@@ -250,7 +250,7 @@ private static int mergeSameBuckets(double[] values, double[] weights, int nextI
 
         int current = 0;
         for (int i = 1; i < nextIndex; i++) {
-            if (values[current] == values[i]) {
+            if (values[current] == values[i] || (Double.isNaN(values[current]) && Double.isNaN(values[i]))) {
                 weights[current] += weights[i];
             }
             else {
@@ -350,7 +350,7 @@ private Entry(int id, double value, double weight, Entry left, Entry right)
                 penalty = computePenalty(value, weight, right.value, right.weight);
             }
             else {
-                penalty = Double.POSITIVE_INFINITY;
+                penalty = Double.NaN;
             }
 
             if (left != null) {

presto-main-base/src/test/java/com/facebook/presto/operator/aggregation/TestNumericHistogram.java

@@ -133,4 +133,21 @@ public void testMergeDifferent()
         histogram1.mergeWith(histogram2);
         assertEquals(histogram1.getBuckets(), expected.getBuckets());
     }
+
+    @Test
+    public void testNaN()
+    {
+        NumericHistogram histogram = new NumericHistogram(2, 100);
+
+        histogram.add(Double.NaN, 1);
+        histogram.add(2, 1);
+        histogram.add(Double.NaN, 1);
+
+        Map<Double, Double> expected = ImmutableMap.<Double, Double>builder()
+                .put(Double.NaN, 2.0)
+                .put(2.0, 1.0)
+                .build();
+
+        assertEquals(histogram.getBuckets(), expected);
+    }
 }