C'è un modo per costruire un HashSet con la funzione initializator in Kotlin?

Question

Per leggere Stelle da un file nel Boomerang Constelations problema Facebook Hacker Cup's 2016, seguente funzione di estensione può essere definita:

fun BufferedReader.readStars(n: Int): Set<Star> { 
    return Array(n) { 
     val (l1, l2) = readLine().split(" ").map { it.toInt() } 
     Star(l1, l2) 
    }.toHashSet() 
} 

codice è compatto ma i valori sono leggere prima in una matrice e poi convertito HashSet. C'è un modo per inizializzare direttamente un HashSet con le dimensioni di n e la funzione initializator in Kotlin?

UPDATE: C'è un esistente modo in librerie standard di Kotlin?

Answer 1

Dal HashSet è una classe java in modo che è possibile solo inizializzarlo in un modo fornito da JDK.

Mentre non c'è alcun metodo di supporto in Kotlin runtime è facile da scrivere da soli in questo modo:

public fun <T> hashSetOf(size: Int, initializer: (Int) -> T): HashSet<T> { 
    val result = HashSet<T>(size) 
    0.rangeTo(size - 1).forEach { 
     result.add(initializer(it)) 
    } 
    return result 
} 

Answer 2

si può sempre usare apply per inizializzare gli oggetti in-place:

HashSet<Star>(n).apply { 
    repeat(n) { 
     val (l1, l2) = readLine()!!.split(' ').map { it.toInt() } 
     put(Star(l1, l2)) 
    } 
} 

Se anche questo è troppo fastidioso digitare ogni volta, scrivere una funzione di estensione:

inline fun <T> createHashSet(n : Int, crossinline fn: (Int) -> T) = HashSet<T>(n).apply { 
    repeat(n) { add(fn(it)) } 
} 

utilizzati:

createHashSet<Star>(n) { 
    val (l1, l2) = readLine()!!.split(' ').map { it.toInt() } 
    Star(l1, l2) 
} 

Answer 3

Come @miensol ha sottolineato HashSet inizializzazione è limitata ai costruttori messi a disposizione dal JDK. Kotlin ha aggiunto una funzione hashSetOf che inizializza uno HashSet vuoto e quindi aggiunge gli elementi specificati ad esso.

Per evitare prima lettura i valori in un array è possibile utilizzare un kotlin.Sequence che è "I valori sono valutati pigramente":

fun BufferedReader.readStars(n: Int): Set<Star> { 
    return lineSequence().take(n).map { 
     val (l1, l2) = it.split(" ").map { it.toInt() } 
     Star(l1, l2) 
    }.toHashSet() 
} 

Answer 4

Sembra che ti stai chiedendo una domanda XY (http://xyproblem.info/). Vuoi davvero sapere come scrivere readStars nel modo più efficiente, ma invece chiedi informazioni su HashSet. Penso che @ mfulton26 risponda alla tua domanda a seconda di cosa viene chiesto.

Ecco la risposta per "come faccio a scrivere questo nel modo più efficiente:"

si hanno due opzioni. In primo luogo, una versione che auto-chiude il flusso alla fine:

fun BufferedReader.readStars(n: Int): Set<Star> { 
    return use { 
     lineSequence().map { line -> 
      val idx = line.indexOf(' ') 
      Star(line.substring(0, idx).toInt(), line.substring(idx + 1).toInt()) 
     }.toSet() 
    } 
} 

E in secondo luogo, una versione che non è così:

fun BufferedReader.readStars(n: Int): Set<Star> { 
    return lineSequence().map { line -> 
      val idx = line.indexOf(' ') 
      Star(line.substring(0, idx).toInt(), line.substring(idx+1).toInt()) 
     }.toSet() 
} 

Né versione crea una matrice, non fanno a fare copie di dati. Trasmettono i dati attraverso una sequenza che crea il Set e lo riempie direttamente.

Altre note

Non c'è bisogno di utilizzare scissione se siete veramente preoccupati per le allocazioni e le prestazioni. Basta usare indexOf(char) e dividere la stringa da soli usando substring.

Se fate una spaccatura, quindi si prega di utilizzare split(char) non split(String) quando si sta cercando di dividere su una char