11

我是 Scala 的新手，通过 Haskell 进行函数式编程的经验非常有限。

我想尝试编写从单个输入列表构造的所有可能对的列表。例子：

val nums = List[Int](1, 2, 3, 4, 5)   // Create an input list
val pairs = composePairs(nums)        // Function I'd like to create

// pairs == List[Int, Int]((1, 1), (1, 2), (1, 3), (1, 4), (1, 5), (2, 1) ... etc)

我尝试zip在每个元素上使用整个列表，希望它会在整个列表中复制一个项目。它不起作用（仅匹配第一个可能的对）。我不确定如何重复一个元素（Haskell 会这样做cycle，take我相信），而且我在遵循 Scala 上的文档时遇到了麻烦。

这让我想，可能有一种更简洁、更实用的方式来获得我想要的结果。有人有好的解决方案吗？

4

3 回答 3

28

这个怎么样：

val pairs = for(x <- nums; y <- nums) yield (x, y)
于 2012-08-03T21:17:07.470 回答
10

对于那些不想重复的人：

val uniquePairs = for {
      (x, idxX) <- nums.zipWithIndex
      (y, idxY) <- nums.zipWithIndex
      if idxX < idxY
    } yield (x, y)

val nums = List(1,2,3,4,5)
uniquePairs: List[(Int, Int)] = List((1, 2), (1, 3), (1, 4), (1, 5), (2, 3), (2, 4), (2, 5), (3, 4), (3, 5), (4, 5))
于 2017-08-17T01:24:28.123 回答
1

这是另一个使用 map 和 flatten 的版本

val pairs = nums.flatMap(x => nums.map(y => (x,y)))

List[(Int, Int)] = List((1,1), (1,2), (1,3), (1,4), (1,5), (2,1), (2,2), (2,3), (2,4), (2,5), (3,1), (3,2), (3,3), (3,4), (3,5), (4,1), (4,2), (4,3), (4,4), (4,5), (5,1), (5,2) (5,3), (5,4), (5,5))

如果您愿意，可以轻松地将其包装到 composePairs 函数中：

def composePairs(nums: Seq[Int]) =
    nums.flatMap(x => nums.map(y => (x,y)))
于 2012-08-03T21:52:38.720 回答