最近在项目中遇到一个需求,需要在一串给定的日期序列中找到给定的日期,如果没有找到相同的就返回与给定日期最接近的日期。
遍历整个数组自然是可以找到,但既然是有序数组当然可以用二分法了,下面上代码:
/**
* 二分法获取最接近给定日期的日期
* @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);
}
}
}
如果是无序数组,可以考虑先排序,变为有序数组后再使用此方法。