Jak uzyskać losowy element z zestawu w Swift?

Question

Od Swift 1.2 (dostępny jako wersja beta) Apple wprowadza typ kolekcji Set.

Say, Mam zestaw jak:

var set = Set<Int>(arrayLiteral: 1, 2, 3, 4, 5) 

Teraz chcę dostać losowy element z niego. Pytanie brzmi: jak? Set nie udostępnia subscript(Int), jak Array. Zamiast tego ma subscript(SetIndex<T>). Ale po pierwsze, SetIndex<T> nie ma dostępnych inicjalizatorów (stąd nie mogę po prostu utworzyć indeksu z potrzebnym offsetem), a po drugie, nawet jeśli mogę uzyskać indeks dla pierwszego elementu w zbiorze (var startIndex = set.startIndex), to jedyny sposób może dostać się do N-tego indeksu przez kolejne połączenia do successor().

Zatem można zobaczyć tylko dwie możliwości w tym momencie, zarówno brzydkie i kosztownych:

• Konwersja zestawu do tablicy (var array = [Int](set)) i dokonuje indeksu dolnego (który idealnie przyjmuje Int); lub
• Pobierz indeks pierwszego elementu w zbiorze, przechodź przez łańcuchy metod successor(), aby uzyskać indeks N-ty, a następnie odczytaj odpowiedni element za pomocą indeksu zestawu.

Czy tęsknię za jakimś innym sposobem?

UPDATE

Jak @rintaro wskazał, należy użyć wcześniej() funkcja natychmiast dostać się do indeksu chcę. Np .:

var set = Set<Int>(arrayLiteral: 1, 2, 3, 4, 5) 
let randomOffset = Int(arc4random_uniform(UInt32(set.count))) 
let random = set[advance(set.startIndex, randomOffset)] 

UPDATE 2

Okazuje się, że advance() ma złożoność O (n), co zasadniczo czyni ją równoważne prostu przechodząc przez łańcuch successor().

Answer 1

Prawdopodobnie najlepszym rozwiązaniem jest advance który idzie successor dla Ciebie:

func randomElementIndex<T>(s: Set<T>) -> T { 
    let n = Int(arc4random_uniform(UInt32(s.count))) 
    let i = advance(s.startIndex, n) 
    return s[i] 
} 

(EDIT: Heh; Zauważyliśmy, że faktycznie aktualizowany kwestię włączenia tę odpowiedź zanim dodałem go do mojej odpowiedzi ... dobrze, nadal jest to dobry pomysł i nauczyłem się czegoś.: D)

Możesz także chodzić po zestawie, a nie w indeksach (to była moja pierwsza myśl, ale potem przypomniałem sobie advance).

func randomElement<T>(s: Set<T>) -> T { 
    let n = Int(arc4random_uniform(UInt32(s.count))) 
    for (i, e) in enumerate(s) { 
     if i == n { return e } 
    } 
    fatalError("The above loop must succeed") 
} 

Answer 2

Jeśli chcesz „random” elementu z Set następnie użyć:

/// A member of the set, or `nil` if the set is empty. 
var first: T? { get } 

Pobierz 0TH indeksu lub indeks 1.000.000 ma znaczenia - wszystkie są arbitralne przedmiot.

Ale, jeśli chcesz powtarzające się połączenia, aby zwrócić za każdym razem inny, prawdopodobnie element, wtedy first może nie pasować do rachunku.

Answer 3

extension Set { 
    func randomElement() -> Element? { 
     return count == 0 ? nil : self[advance(self.startIndex, Int(arc4random()) % count)] 
    } 
} 

Answer 4

Według powyższych uwag re Swift aktualizacji, używany niewielką zmianę o przedłużenie do ustaw:

func randomElement() -> Element? 
{ 
    let randomInt = Int(arc4random_uniform(UInt32(self.count))) 
    let index = startIndex.advancedBy(randomInt) 
    return count == 0 ? nil: self[index] 
} 

Answer 5

w Swift 3

extension Set { 
    public func randomObject() -> Element? { 
     let n = Int(arc4random_uniform(UInt32(self.count))) 
     let index = self.index(self.startIndex, offsetBy: n) 
     return self.count > 0 ? self[index] : nil 
    } 
}