首先,我们需要对日期进行排序。在 ruby 中,这很简单
sorted_dates = dates.sort
如果您知道日期已排序,则只需从第一个日期开始,并在您遍历日期范围时递增一个。如果您的数组中的下一个日期不是您预期的日期,请将缺少的日期添加到您的 missing_dates 数组中,并继续递增直到您到达包含的日期。
此代码可能如下所示:
def find_missing_dates(sorted_dates)
current_date = sorted_dates[0]
missing_dates = Set.new
sorted_dates.each do |date|
while current_date != date
missing_dates << current_date
current_date += 1.day
end
current_date += 1.day
end
end
对于一般情况,这是 O(N),因此为了提高效率,我们可以一分为二并递归。
def dates_between(lower, upper)
(lower..upper).to_a - [lower,upper]
end
def find_missing_dates(sorted_dates, missing_dates = Set.new)
min_date = sorted_dates[0]
max_date = sorted_dates[-1]
if (min_date - max_date).to_i == (sorted_dates.count - 1)
missing_dates
else
middle_date_lower = sorted_dates[sorted_dates.count / 2 - 1]
middle_date_upper = sorted_dates[sorted_dates.count / 2]
unless (middle_date_upper - middle_date_lower) == 1
missing_dates.merge(dates_between(middle_date_lower, middle_date_upper))
end
find_missing_dates(sorted_dates[0..(sorted_dates.count/2 - 1)], missing_dates).merge(find_missing_dates(sorted_dates[(sorted_dates.count/2)..-1]))
end
end
find_missing_dates(sorted_dates)
这仍然是最坏情况 O(N),但平均情况是 O(log N)