Какой лучший способ перемешать массив NSMutableArray? - PullRequest
185 голосов
/ 11 сентября 2008

Если у вас есть NSMutableArray, как вы случайно перемешиваете элементы?

(у меня есть свой ответ на этот вопрос, который опубликован ниже, но я новичок в Какао, и мне интересно узнать, есть ли лучший способ.)


Обновление: как отмечает @Mukesh, начиная с iOS 10+ и macOS 10.12+, существует метод -[NSMutableArray shuffledArray], который можно использовать для перемешивания. Подробнее см. https://developer.apple.com/documentation/foundation/nsarray/1640855-shuffledarray?language=objc. (Но учтите, что это создает новый массив, а не перетасовывает элементы на месте.)

Ответы [ 12 ]

345 голосов
/ 11 сентября 2008

Я решил это, добавив категорию в NSMutableArray.

Редактировать: Удален ненужный метод благодаря ответу Ladd.

Редактировать: Изменено (arc4random() % nElements) на arc4random_uniform(nElements) благодаря ответу Григория Гольцова и комментариям Михо и Бладиблы

Редактировать: Улучшение цикла, благодаря комментарию Рона

Редактировать: Добавлена ​​проверка, что массив не пуст, благодаря комментарию Махеша Агравала

//  NSMutableArray_Shuffling.h

#if TARGET_OS_IPHONE
#import <UIKit/UIKit.h>
#else
#include <Cocoa/Cocoa.h>
#endif

// This category enhances NSMutableArray by providing
// methods to randomly shuffle the elements.
@interface NSMutableArray (Shuffling)
- (void)shuffle;
@end


//  NSMutableArray_Shuffling.m

#import "NSMutableArray_Shuffling.h"

@implementation NSMutableArray (Shuffling)

- (void)shuffle
{
    NSUInteger count = [self count];
    if (count <= 1) return;
    for (NSUInteger i = 0; i < count - 1; ++i) {
        NSInteger remainingCount = count - i;
        NSInteger exchangeIndex = i + arc4random_uniform((u_int32_t )remainingCount);
        [self exchangeObjectAtIndex:i withObjectAtIndex:exchangeIndex];
    }
}

@end
74 голосов
/ 12 сентября 2008

Вам не нужен метод swapObjectAtIndex. exchangeObjectAtIndex: withObjectAtIndex: уже существует.

38 голосов
/ 04 июня 2012

Поскольку я пока не могу комментировать, я думал, что внесу полный ответ. Я изменил реализацию Кристофера Джонсона для своего проекта несколькими способами (действительно пытаясь сделать его как можно более кратким), одним из них было arc4random_uniform(), потому что оно избегает смещения по модулю .

// NSMutableArray+Shuffling.h
#import <Foundation/Foundation.h>

/** This category enhances NSMutableArray by providing methods to randomly
 * shuffle the elements using the Fisher-Yates algorithm.
 */
@interface NSMutableArray (Shuffling)
- (void)shuffle;
@end

// NSMutableArray+Shuffling.m
#import "NSMutableArray+Shuffling.h"

@implementation NSMutableArray (Shuffling)

- (void)shuffle
{
    NSUInteger count = [self count];
    for (uint i = 0; i < count - 1; ++i)
    {
        // Select a random element between i and end of array to swap with.
        int nElements = count - i;
        int n = arc4random_uniform(nElements) + i;
        [self exchangeObjectAtIndex:i withObjectAtIndex:n];
    }
}

@end
9 голосов
/ 15 ноября 2016

В iOS 10 вы можете использовать новый shuffled API:

https://developer.apple.com/reference/foundation/nsarray/1640855-shuffled

let shuffledArray = array.shuffled()
8 голосов
/ 21 ноября 2015

Немного улучшенное и краткое решение (по сравнению с основными ответами).

Алгоритм такой же и описан в литературе как " Fisher-Yates shuffle ".

В Objective-C:

@implementation NSMutableArray (Shuffle)
// Fisher-Yates shuffle
- (void)shuffle
{
    for (NSUInteger i = self.count; i > 1; i--)
        [self exchangeObjectAtIndex:i - 1 withObjectAtIndex:arc4random_uniform((u_int32_t)i)];
}
@end

В Swift 3.2 и 4.x:

extension Array {
    /// Fisher-Yates shuffle
    mutating func shuffle() {
        for i in stride(from: count - 1, to: 0, by: -1) {
            swapAt(i, Int(arc4random_uniform(UInt32(i + 1))))
        }
    }
}

В Swift 3.0 и 3.1:

extension Array {
    /// Fisher-Yates shuffle
    mutating func shuffle() {
        for i in stride(from: count - 1, to: 0, by: -1) {
            let j = Int(arc4random_uniform(UInt32(i + 1)))
            (self[i], self[j]) = (self[j], self[i])
        }
    }
}

Примечание: Более краткое решение в Swift возможно с iOS10, используя GameplayKit.

Примечание: Доступен также алгоритм нестабильной перетасовки (с вынужденным изменением всех позиций, если число> 1)

6 голосов
/ 19 августа 2009

Это самый простой и быстрый способ перемешивания NSArrays или NSMutableArrays. (object puzzles - это NSMutableArray, он содержит объекты головоломки. Я добавил в индекс переменной объекта головоломки, который указывает начальную позицию в массиве)

int randomSort(id obj1, id obj2, void *context ) {
        // returns random number -1 0 1
    return (random()%3 - 1);    
}

- (void)shuffle {
        // call custom sort function
    [puzzles sortUsingFunction:randomSort context:nil];

    // show in log how is our array sorted
        int i = 0;
    for (Puzzle * puzzle in puzzles) {
        NSLog(@" #%d has index %d", i, puzzle.index);
        i++;
    }
}

вывод журнала:

 #0 has index #6
 #1 has index #3
 #2 has index #9
 #3 has index #15
 #4 has index #8
 #5 has index #0
 #6 has index #1
 #7 has index #4
 #8 has index #7
 #9 has index #12
 #10 has index #14
 #11 has index #16
 #12 has index #17
 #13 has index #10
 #14 has index #11
 #15 has index #13
 #16 has index #5
 #17 has index #2

вы также можете сравнить obj1 с obj2 и решить, что вы хотите вернуть Возможные значения:

  • NSOrderedAscending = -1
  • NSOrderedSame = 0
  • NSOrderedDescending = 1
2 голосов
/ 05 мая 2017

В iOS 10 вы можете использовать NSArray shuffled() из GameplayKit . Вот помощник для Array в Swift 3:

import GameplayKit

extension Array {
    @available(iOS 10.0, macOS 10.12, tvOS 10.0, *)
    func shuffled() -> [Element] {
        return (self as NSArray).shuffled() as! [Element]
    }
    @available(iOS 10.0, macOS 10.12, tvOS 10.0, *)
    mutating func shuffle() {
        replaceSubrange(0..<count, with: shuffled())
    }
}
2 голосов
/ 13 сентября 2013

Существует хорошая популярная библиотека, в которой этот метод является частью, называемой SSToolKit в GitHub . Файл NSMutableArray + SSToolkitAdditions.h содержит метод shuffle. Вы можете использовать это также. Среди этого, кажется, тонны полезных вещей.

Главная страница этой библиотеки здесь .

Если вы используете это, ваш код будет выглядеть так:

#import <SSCategories.h>
NSMutableArray *tableData = [NSMutableArray arrayWithArray:[temp shuffledArray]];

В этой библиотеке также есть Pod (см. CocoaPods)

1 голос
/ 21 марта 2014

Если элементы повторяются.

например. массив: A A A B B или B B A A A

Единственное решение: A B A B A

sequenceSelected - это NSMutableArray, который хранит элементы класса obj, которые являются указателями на некоторую последовательность.

- (void)shuffleSequenceSelected {
    [sequenceSelected shuffle];
    [self shuffleSequenceSelectedLoop];
}

- (void)shuffleSequenceSelectedLoop {
    NSUInteger count = sequenceSelected.count;
    for (NSUInteger i = 1; i < count-1; i++) {
        // Select a random element between i and end of array to swap with.
        NSInteger nElements = count - i;
        NSInteger n;
        if (i < count-2) { // i is between second  and second last element
            obj *A = [sequenceSelected objectAtIndex:i-1];
            obj *B = [sequenceSelected objectAtIndex:i];
            if (A == B) { // shuffle if current & previous same
                do {
                    n = arc4random_uniform(nElements) + i;
                    B = [sequenceSelected objectAtIndex:n];
                } while (A == B);
                [sequenceSelected exchangeObjectAtIndex:i withObjectAtIndex:n];
            }
        } else if (i == count-2) { // second last value to be shuffled with last value
            obj *A = [sequenceSelected objectAtIndex:i-1];// previous value
            obj *B = [sequenceSelected objectAtIndex:i]; // second last value
            obj *C = [sequenceSelected lastObject]; // last value
            if (A == B && B == C) {
                //reshufle
                sequenceSelected = [[[sequenceSelected reverseObjectEnumerator] allObjects] mutableCopy];
                [self shuffleSequenceSelectedLoop];
                return;
            }
            if (A == B) {
                if (B != C) {
                    [sequenceSelected exchangeObjectAtIndex:i withObjectAtIndex:count-1];
                } else {
                    // reshuffle
                    sequenceSelected = [[[sequenceSelected reverseObjectEnumerator] allObjects] mutableCopy];
                    [self shuffleSequenceSelectedLoop];
                    return;
                }
            }
        }
    }
}
0 голосов
/ 09 февраля 2015

Редактировать: Это не правильно. В справочных целях я не удалил этот пост. См. Комментарии о причине, почему этот подход не является правильным.

Простой код здесь:

- (NSArray *)shuffledArray:(NSArray *)array
{
    return [array sortedArrayUsingComparator:^NSComparisonResult(id obj1, id obj2) {
        if (arc4random() % 2) {
            return NSOrderedAscending;
        } else {
            return NSOrderedDescending;
        }
    }];
}
...