用二分法在有序数组中快速查找与给定值最接近的值
糊涂粥 于 2024-06-08 08:42:26

最近在项目中遇到一个需求,需要在一串给定的日期序列中找到给定的日期,如果没有找到相同的就返回与给定日期最接近的日期。

遍历整个数组自然是可以找到,但既然是有序数组当然可以用二分法了,下面上代码:

    /**
     * 二分法获取最接近给定日期的日期
     * @param string $date 给定的日期
     * @param array $dates 有序日期数组
     * @return string
     */
    protected function findClosest(string $date, array $dates): string
    {
        if (in_array($date, $dates)) {
            return $date;  // 存在指定值直接返回结果
        } else {
            if ($date > max($dates)) return max($dates); // 大于最大值直接返回最大值
            if ($date < min($dates)) return min($dates); // 小于最小值直接返回最小值
            // 找中点
            $mid1 = ceil(count($dates) * .5 - 1);
            $mid2 = $mid1 + 1;
            if ($date > min($dates[$mid1], $dates[$mid2]) && $date < max($dates[$mid1], $dates[$mid2])) {
                // 在中点之间,则比较差的绝对值找到最接近的值
                $abs1 = abs($date - $dates[$mid1]);
                $abs2 = abs($date - $dates[$mid2]);
                return $abs1 < $abs2 ? $dates[$mid1] : max($dates[$mid2], $dates[$mid1]);
            } else {
                // 未在中点之间,判断所在区间并截短数组递归查找
                $dates = $date > min($dates[0], $dates[$mid1]) && $date < max($dates[0], $dates[$mid1]) ?
                    array_splice($dates, 0, $mid1 + 1) :
                    array_splice($dates, $mid1 + 1);
                return $this->findClosest($date, $dates);
            }
        }
    }

如果是无序数组,可以考虑先排序,变为有序数组后再使用此方法。

阅读 (2297)