2011-01-21 17 views

risposta

15

Questo è il trucco che io di solito uso:

def split[T](list: List[T]) : List[List[T]] = list match { 
    case Nil => Nil 
    case h::t => val segment = list takeWhile {h ==} 
    segment :: split(list drop segment.length) 
} 

In realtà ... Non è, di solito astratta sul tipo di raccolta e ottimizzare con la coda ricorsione pure, ma voluto mantenere la semplice risposta.

+2

Bello. Ma questo non sarebbe abbastanza comune da giustificare la sua funzione di libreria? –

+0

Nessun argomento, è sicuramente presente in più di una domanda ... –

+0

@KevinWright "ottimizza anche con la ricorsione della coda" -> Come faresti? –

8
val xs = List(5, 2, 3, 3, 3, 5, 5, 3, 3, 2, 2, 2) 

Ecco un altro modo.

(List(xs.take(1)) /: xs.tail)((l,r) => 
    if (l.head.head==r) (r :: l.head) :: l.tail else List(r) :: l 
).reverseMap(_.reverse) 
6
list.foldRight(List[List[Int]]()){ 
    (e, l) => l match { 
    case (`e` :: xs) :: fs => (e :: e :: xs) :: fs 
    case _ => List(e) :: l 
    } 
} 

O

list.zip(false :: list.sliding(2).collect{case List(a,b) => a == b}.toList) 
.foldLeft(List[List[Int]]())((l,e) => if(e._2) (e._1 :: l.head) :: l.tail 
             else List(e._1) :: l).reverse 

[Edit]

//find the hidden way 
//the beauty must be somewhere 
//when we talk scala 

def split(l: List[Int]): List[List[Int]] = 
    l.headOption.map{x => val (h,t)=l.span{x==}; h::split(t)}.getOrElse(Nil) 
+0

Qualcuno dei metodi avrebbe funzionato con i valori NaN? Sto cercando di capire se vedo n valori Double.NaN consecutivi in ​​un vettore.Guardando le soluzioni pubblicate qui, sembra che questo non è niente che mi verrà fuori da solo. – TomTom101

+0

I valori NaN richiederebbero una gestione speciale, in effetti. Forse potresti racchiudere il 'Int's in' Option [Int] 'e convertire' NaN's in 'None's. Quindi le solite soluzioni basate sull'uguaglianza funzionerebbero. O potresti scrivere la tua funzione di uguaglianza. – Landei

8

Dannazione Rex Kerr, per scrivere la risposta che mi piacerebbe andare per. Dato che ci sono differenze stilistiche minori, ecco il mio introito:

list.tail.foldLeft(List(list take 1)) { 
    case (acc @ (lst @ hd :: _) :: tl, el) => 
     if (el == hd) (el :: lst) :: tl 
     else (el :: Nil) :: acc 
} 

Poiché gli elementi sono identici, non mi sono preoccupato invertendo le liste parziali.

+0

Puoi spiegare (o indicarmi il link) case (acc @ (lst @ hd :: _). Non ho mai visto prima questa sintassi. Grazie in anticipo. – gashu

+1

@gashu In un pattern matching, 'x @ y' significa che 'x' sarà assegnato a qualunque cosa è _matched_ di' y'. Quindi, ad esempio, 'x @ _ :: _' assegnerà a x una lista non vuota (cioè una lista con una testa e una coda , come corrisponde a '_ :: _'). Quindi' acc', sopra, è l'intera lista, e 'lst' è il capo della lista. –

+0

@ daniel-c-sobral così tali espressioni sono usate solo nel pattern corrispondenza, o in un altro contesto, significano qualcosa di diverso? – gashu

5

Ho queste implementazioni in giro da lavorare sui metodi di raccolta. Alla fine ho controllato in più semplici implementazioni di inits e tails e lasciato il cluster. Ogni nuovo metodo, non importa quanto semplice, finisca per raccogliere una grande tassa che è difficile vedere dall'esterno. Ma ecco l'implementazione che non ho usato.

import generic._ 
import scala.reflect.ClassManifest 
import mutable.ListBuffer 
import annotation.tailrec 
import annotation.unchecked.{ uncheckedVariance => uV } 

def inits: List[Repr] = repSequence(x => (x, x.init), Nil) 
def tails: List[Repr] = repSequence(x => (x, x.tail), Nil) 
def cluster[A1 >: A : Equiv]: List[Repr] = 
    repSequence(x => x.span(y => implicitly[Equiv[A1]].equiv(y, x.head))) 

private def repSequence(
    f: Traversable[A @uV] => (Traversable[A @uV], Traversable[A @uV]), 
    extras: Traversable[A @uV]*): List[Repr] = { 

    def mkRepr(xs: Traversable[A @uV]): Repr = newBuilder ++= xs result 
    val bb = new ListBuffer[Repr] 

    @tailrec def loop(xs: Repr): List[Repr] = { 
    val seq = toCollection(xs) 
    if (seq.isEmpty) 
     return (bb ++= (extras map mkRepr)).result 

    val (hd, tl) = f(seq) 
    bb += mkRepr(hd) 
    loop(mkRepr(tl)) 
    } 

    loop(self.repr) 
} 

[Modifica: Ho dimenticato che altre persone non conoscono l'interno. Questo codice è stato scritto da dentro di TraversableLike, quindi non sarebbe correre fuori dalla scatola]

0

questo potrebbe essere più semplice:.

val input = List(5, 2, 3, 3, 3, 5, 5, 3, 3, 2, 2, 2) 
input groupBy identity values 
+9

Questo raggrupperà tutti gli elementi identici, non solo sequenziali. –

2

Ecco una soluzione ricorsiva in coda ispirato @ Kevin Wright e @ Landei:

@tailrec 
def sliceEqual[A](s: Seq[A], acc: Seq[Seq[A]] = Seq()): Seq[Seq[A]] = { 
    s match { 
    case fst :: rest => 
     val (l, r) = s.span(fst==) 
     sliceEqual(r, acc :+ l) 
    case Nil => acc 
    } 
} 
Problemi correlati