Как я могу получить пересечение, объединение и подмножество массивов в Ruby? - PullRequest
148 голосов
/ 15 апреля 2011

Я хочу создать различные методы для класса с именем Multiset .

У меня есть все необходимые методы, но я не уверен, как писать методы пересечения, объединения и подмножества.

Для пересечения и объединения мой код начинается так:

def intersect(var)
  x = Multiset.new
end

Вот пример:

X = [1, 1, 2, 4]
Y = [1, 2, 2, 2]

тогда пересечение X и Y равно [1, 2].

Ответы [ 3 ]

288 голосов
/ 15 апреля 2011

Полагаю, X и Y - это массивы?Если это так, есть очень простой способ сделать это:

x = [1, 1, 2, 4]
y = [1, 2, 2, 2]

# intersection
x & y            # => [1, 2]

# union
x | y            # => [1, 2, 4]

# difference
x - y            # => [4]

Источник

133 голосов
/ 15 апреля 2011

Используя тот факт, что вы можете выполнять операции над множествами, выполняя & (пересечение), - (различие) и | (объединение).

Очевидно, я не реализовалMultiSet to spec, но это должно помочь вам начать:

class MultiSet
  attr_accessor :set
  def initialize(set)
    @set = set
  end
  # intersection
  def &(other)
    @set & other.set
  end
  # difference
  def -(other)
    @set - other.set
  end
  # union
  def |(other)
    @set | other.set
  end
end

x = MultiSet.new([1,1,2,2,3,4,5,6])
y = MultiSet.new([1,3,5,6])

p x - y # [2,2,4]
p x & y # [1,3,5,6]
p x | y # [1,2,3,4,5,6]
3 голосов
/ 26 декабря 2018

Если Multiset расширяется от Array класса

x = [1, 1, 2, 4, 7]
y = [1, 2, 2, 2]
z = [1, 1, 3, 7]

UNION

x.union(y)           # => [1, 2, 4, 7]      (ONLY IN RUBY 2.6)
x.union(y, z)        # => [1, 2, 4, 7, 3]   (ONLY IN RUBY 2.6)
x | y                # => [1, 2, 4, 7]

РАЗНИЦА

x.difference(y)      # => [4, 7] (ONLY IN RUBY 2.6)
x.difference(y, z)   # => [4] (ONLY IN RUBY 2.6)
x - y                # => [4, 7]

INTERSECTION

x & y                # => [1, 2]

Для получения дополнительной информации о новых методах в Ruby 2.6, вы можете проверить это сообщение в блоге о его новых функциях

...