2012-06-21 26 views
5

Biorąc pod uwagę List takich jakPodział lista scala na n list przeplatania

List(1, 2, 3, 4, 5, 6, 7) 

co jest najlepszym sposobem, aby podzielić go na n podlist, umieszczanie elementów w każdej z list w sposób cykliczny?

np. jeśli n = 3, wynik powinien być

List(List(1, 4, 7), List(2, 5), List(3, 6)) 

Myślałem, że będzie to metoda w zbiorach API to zrobić, ale nie wydaje się go znaleźć.

punkty za klasą jednej wkładki;)

+0

podobne pytanie, za pomocą strumieni: http: // stackoverflow .pl/questions/17115345/split-a-stream-in- – AmigoNico

Odpowiedz

10
scala> def round[T](l: List[T], n: Int) = (0 until n).map{ i => l.drop(i).sliding(1, n).flatten.toList }.toList 
round: [T](l: List[T], n: Int)List[List[T]] 

scala> round((1 to 7).toList, 3) 
res4: List[List[Int]] = List(List(1, 4, 7), List(2, 5), List(3, 6)) 
4

Oto prosty jedno-liner:

scala> List.range(1, 10) 
res11: List[Int] = List(1, 2, 3, 4, 5, 6, 7, 8, 9) 

scala> res11.grouped(3).toList.transpose 
res12: List[List[Int]] = List(List(1, 4, 7), List(2, 5, 8), List(3, 6, 9)) 

Niestety to nie będzie działać, gdy lista nie jest transposeable .

scala> List.range(1, 8).grouped(3).toList.transpose 
java.lang.IllegalArgumentException: transpose requires all collections have the 
same size 

Możesz skorzystać z następujących metod, aby to transposeable, a następnie zastosować powyższe podejście.

scala> def extend[A](xs: List[A], c: Int): List[Option[A]] = { 
    | val n = Stream.iterate(c)(c +).find(_ >= xs.length).get 
    | xs.map(Some(_)).padTo(n, None) 
    | } 
extend: [A](xs: List[A], c: Int)List[Option[A]] 

scala> List.range(1, 8) 
res13: List[Int] = List(1, 2, 3, 4, 5, 6, 7) 

scala> extend(res13, 3).grouped(3).toList.transpose.map(_.flatten) 
res14: List[List[Int]] = List(List(1, 4, 7), List(2, 5), List(3, 6)) 

ułożyła:

scala> def round[A](xs: List[A], c: Int) = { 
    | val n = Stream.iterate(c)(c +).find(_ >= xs.length).get 
    | val ys = xs.map(Some(_)).padTo(n, None) 
    | ys.grouped(c).toList.transpose.map(_.flatten) 
    | } 
round: [A](xs: List[A], c: Int)List[List[A]] 

scala> round(List.range(1, 10), 3) 
res16: List[List[Int]] = List(List(1, 4, 7), List(2, 5, 8), List(3, 6, 9)) 

scala> round(List.range(1, 8), 3) 
res17: List[List[Int]] = List(List(1, 4, 7), List(2, 5), List(3, 6)) 
+0

'ys.grouped (3)' czy to literówka? – senia

+0

@senia, dlaczego to byłby literówka? – missingfaktor

+0

powinno być 'ys.grouped (c)' – senia

0

Jeśli nie obchodzi co czyni go round robin można zrobić:

val list = List(1, 2, 3, 4, 5, 6, 7) 
list.grouped(Math.ceil(list.size/3))