Получить подмножества различных размеров в Scala? - PullRequest
Новая
       24

Получить подмножества различных размеров в Scala?

1 голос
/ 29 июня 2019

Я могу получить подмножества определенного размера или все из них. 

scala> println((1 to 25).toSet.subsets.size)
33554432
scala> println((1 to 25).toSet.subsets(1).size)
25
scala> println((1 to 25).toSet.subsets(10).size)
1

Чтобы получить только подмножества размера, скажем, 10 или меньше, я мог бы вернуться к циклу и взять объединение.Каким будет функциональный способ программирования?

Ответы [ 2 ]

3 голосов
/ 29 июня 2019

Вы можете использовать flatMap следующим образом: 

(1 to 10).flatMap((1 to 25).toSet.subsets(_))

Например, ниже приведен список всех subsets размером до 3 в Set из 6 элементов: 

(1 to 3).flatMap((1 to 6).toSet.subsets(_))
// res1: scala.collection.immutable.IndexedSeq[scala.collection.immutable.Set[Int]] = Vector(
//   Set(5), Set(1), Set(6), Set(2), Set(3), Set(4),
//   Set(5, 1), Set(5, 6), Set(5, 2), Set(5, 3), Set(5, 4), Set(1, 6),
//   Set(1, 2), Set(1, 3), Set(1, 4), Set(6, 2), Set(6, 3), Set(6, 4),
//   Set(2, 3), Set(2, 4), Set(3, 4),
//   Set(5, 1, 6), Set(5, 1, 2), Set(5, 1, 3), Set(5, 1, 4), Set(5, 6, 2),
//   Set(5, 6, 3), Set(5, 6, 4), Set(5, 2, 3), Set(5, 2, 4), Set(5, 3, 4),
//   Set(1, 6, 2), Set(1, 6, 3), Set(1, 6, 4), Set(1, 2, 3), Set(1, 2, 4),
//   Set(1, 3, 4), Set(6, 2, 3), Set(6, 2, 4), Set(6, 3, 4), Set(2, 3, 4)
// )
1 голос
/ 29 июня 2019

Я думаю, это то, что вы ищете. (Использование меньших чисел для визуальной проверки.) 

(2 to 4).flatMap(Set(1 to 5:_*).subsets)
//res0 = Vector(Set(5, 1), Set(5, 2), Set(5, 3), Set(5, 4), Set(1, 2), Set(1, 3), Set(1, 4), Set(2, 3), Set(2, 4), Set(3, 4), Set(5, 1, 2), Set(5, 1, 3), Set(5, 1, 4), Set(5, 2, 3), Set(5, 2, 4), Set(5, 3, 4), Set(1, 2, 3), Set(1, 2, 4), Set(1, 3, 4), Set(2, 3, 4), Set(5, 1, 2, 3), Set(5, 1, 2, 4), Set(5, 1, 3, 4), Set(5, 2, 3, 4), Set(1, 2, 3, 4))
...