DouKaku?

2個以上のリストlist1, list2, list3...が与えられたときに、 その複数個のリストの中の要素を一つずつとりだして組にする方法の全通りのリストを返すコードを書いてください。

Pythonで表現すると下のようになります。

>>> c = CrossProduct([1,2,3,4], "abc")
>>> list(c.all())
[[1, 'a'], [1, 'b'], [1, 'c'], [2, 'a'],
 [2, 'b'], [2, 'c'], [3, 'a'], [3, 'b'],
 [3, 'c'], [4, 'a'], [4, 'b'], [4, 'c']]

>>> c = CrossProduct([0, 1], "ab", ["Foo", "Bar"])
>>> list(c.all())
[[0, 'a', 'Foo'], [0, 'a', 'Bar'], [0, 'b', 'Foo'], [0, 'b', 'Bar']
 [1, 'a', 'Foo'], [1, 'a', 'Bar'], [1, 'b', 'Foo'], [1, 'b', 'Bar']]

順番はこの通りでなくても構いません。返すものはリストと書きましたが、 なんらかの「一度に全部をメモリ上に作成しないリスト状のモノ」がある言語ではそちらを使う方がおすすめです。 数値や文字列を一つのリストに混在させるのがやっかいな言語では整数のリストに限定しても構いません。

このお題はZIGOROuさんとのやりとりにヒントを得て作りました。 (しまった、先にブログで公開されてしまった→Yet Another Hackadelic - 直積の導出と考えうる全ての値を網羅したハッシュの生成)

追記：サンプル出力が間違っていたのでoceanさんの解答を使って出力し直しました。

結果は（見易さのために改行を足しましたが）こんな感じです。

scala> crossproduct(List(1,2,3), List(4,5,6))
res14: List[List[Int]] = 
  List(List(1, 4), List(1, 5), List(1, 6), 
       List(2, 4), List(2, 5), List(2, 6), 
       List(3, 4), List(3, 5), List(3, 6))

scala> crossproduct(List(1,2),List(4,5),List(6,7))
res15: List[List[Int]] = 
  List(List(1, 4, 6), List(1, 4, 7), List(1, 5, 6), List(1, 5, 7), 
       List(2, 4, 6), List(2, 4, 7), List(2, 5, 6), List(2, 5, 7))

1
2
3
4
5
6

def crossproduct[T](xss: List[T]*) = {
  var ps = List(List[T]())
  for (xs <- xss.reverse)
    ps = for (i <- xs; j <- ps) yield i::j
  ps
}

1行にしてみました。

1
2
3
4
5
6
7

def crossProduct(as:List[Any]*) = {
  as.reverse./:(List(List[Any]())){for(i <-_; j <-_) yield j::i}
}


println(crossProduct(List(1,2,3,4), List('a', 'b', 'c')))
println(crossProduct(List(0,1), List('a', 'b'), List("Foo", "Bar")))