在Ruby中计算两个日期范围数组的交集

Question

在Ruby中计算两个日期范围数组的交集

3

给定两个大范围数组...

A = [0..23, 30..53, 60..83, 90..113]
B = [-Float::INFINITY..13, 25..33, 45..53, 65..73, 85..93]

当我进行逻辑连接logical conjuction时...

C = A.mask(B)

然后我期望

describe "Array#mask" do
  it{expect(C = A.mask(B)).to eq([0..13, 30..33, 45..53, 65..73, 90..93])}
end

感觉应该是这样的...

C = A & B
=> []

但是这是空的因为没有一个范围是相同的。

这里有一个图示例。

Logical conjuction waveform .

我已经在范围内包含了无限大，因为解决这个问题通常涉及将范围转换为数组或集合。 我的当前解决方案 这是我的当前解决方案，通过速度和准确性测试。我正在寻找评论和/或建议改进。第二个测试使用优秀的IceCube gem来生成日期范围数组。在我的掩码方法中有一个隐含的假设，即每个计划中的日期范围出现不重叠。

require 'pry'
require 'rspec'
require 'benchmark'
require 'chronic'
require 'ice_cube'
require 'active_support'
require 'active_support/core_ext/numeric'
require 'active_support/core_ext/date/calculations'

A = [0..23, 30..53, 60..83, 90..113]
B = [-Float::INFINITY..13, 25..33, 45..53, 65..73, 85..93]

class Array
  def mask(other)
    a_down = self.map{|r| [:a, r.max]}
    a_up = self.map{|r| [:a, r.min]}

    b_down = other.map{|r| [:b, r.max]}
    b_up = other.map{|r| [:b, r.min]}

    up = a_up + b_up
    down = a_down + b_down

    a, b, start, result = false, false, nil, []
    ticks = (up + down).sort_by{|i| i[1]}
    ticks.each do |tick|
      tick[0] == :a ? a = !a : b = !b
      result << (start..tick[1]) if !start.nil?
      start = a & b ? tick[1] : nil
    end
    return result
  end
end

describe "Array#mask" do
  context "simple integer array" do
    it{expect(C = A.mask(B)).to eq([0..13, 30..33, 45..53, 65..73, 90..93])}
  end

  context "larger date ranges from IceCube schedule" do
    it "should take less than 0.1 seconds" do
      year = Time.now..(Time.now + 52.weeks)
      non_premium_schedule = IceCube::Schedule.new(Time.at(0)) do |s|
        s.duration = 12.hours
        s.add_recurrence_rule IceCube::Rule.weekly.day(:monday, :tuesday, :wednesday, :thursday, :friday).hour_of_day(7).minute_of_hour(0)
      end
      rota_schedule = IceCube::Schedule.new(Time.at(0)) do |s|
        s.duration = 7.hours
        s.add_recurrence_rule IceCube::Rule.weekly(2).day(:tuesday).hour_of_day(15).minute_of_hour(30)
      end
      np = non_premium_schedule.occurrences_between(year.min, year.max).map{|d| d..d+non_premium_schedule.duration}
      rt = rota_schedule.occurrences_between(year.min, year.max).map{|d| d..d+rota_schedule.duration}
      expect(Benchmark.realtime{np.mask(rt)}).to be < 0.1
    end
  end
end

感觉用Ruby现有的核心方法无法做到这一点？我是错过了什么吗？我经常需要计算范围交集。

我也想到了，你可以使用相同的方法通过传递单个项数组来查找两个单一范围之间的交集。例如：

[(54..99)].mask[(65..120)]

我意识到我已经回答了自己的问题，但我觉得把它留在这里作为其他人的参考。

- Kevin Monk

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Jacob Brown · Accepted Answer

我不确定我是否真正理解了你的问题；我有点困惑你的 expect 语句，并且我不知道为什么你的数组大小不同。话虽如此，如果你想计算两个范围的交集，我喜欢这个猴子补丁（来自Ruby: intersection between two ranges）：

class Range
  def intersection(other)
    return nil if (self.max < other.begin or other.max < self.begin) 
    [self.begin, other.begin].max..[self.max, other.max].min
  end
  alias_method :&, :intersection
end

然后你可以这样做：

A = [0..23, 30..53, 60..83, 0..0, 90..113]
B = [-Float::INFINITY..13, 25..33, 45..53, 65..73, 85..93]

A.zip(B).map { |x, y| x & y }
# => [0..13, 30..33, nil, nil, 90..93]

这似乎是一个合理的结果...

编辑

如果你按照上面的方式对Range进行了monkeypatch，并且执行以下操作：

# your initial data
A = [0..23, 30..53, 60..83, 90..113]
B = [-Float::INFINITY..13, 25..33, 45..53, 65..73, 85..93]

A.product(B).map {|x, y| x & y }.compact
# => [0..13, 30..33, 45..53, 65..73, 90..93]

您将获得您指定的结果。我不知道它在性能方面如何比较，并且对排序顺序也不确定...