Meilleure façon de supprimer d'un NSMutableArray tout en itérant ?

Question

En Cocoa, si je veux parcourir un NSMutableArray et supprimer plusieurs objets qui répondent à certains critères, quelle est la meilleure façon de le faire sans redémarrer la boucle à chaque fois que je supprime un objet?

Merci,

Éditer : Juste pour clarifier - je cherchais la meilleure façon, par exemple quelque chose de plus élégant que de mettre à jour manuellement l'index où je me trouve. Par exemple en C++, je peux faire ;

iterator it = someList.begin();

while (it != someList.end())
{
    if (shouldRemove(it))   
        it = someList.erase(it);
}

Answer 1

Pour plus de clarté, j'aime faire une boucle initiale où je rassemble les éléments à supprimer. Ensuite, je les supprime. Voici un exemple utilisant la syntaxe Objective-C 2.0 :

NSMutableArray *discardedItems = [NSMutableArray array];

for (SomeObjectClass *item in originalArrayOfItems) {
    if ([item shouldBeDiscarded])
        [discardedItems addObject:item];
}

[originalArrayOfItems removeObjectsInArray:discardedItems];

Ensuite, il n'y a aucune question sur le bon fonctionnement de la mise à jour des indices, ou sur d'autres petits détails de gestion.

Édité pour ajouter :

Il a été noté dans d'autres réponses que la formulation inverse pourrait être plus rapide. Par exemple, si vous parcourez le tableau et composez un nouveau tableau d'objets à garder, au lieu d'objets à supprimer. Cela peut être vrai (mais qu'en est-il du coût en mémoire et en traitement de l'allocation d'un nouveau tableau, et de la suppression de l'ancien ?), mais même si c'est plus rapide, ce n'est peut-être pas aussi important que pour une implémentation naïve, car les NSArrays ne se comportent pas comme des tableaux "normaux". Ils parlent le langage mais marchent différemment. Voir une bonne analyse ici :

La formulation inverse peut être plus rapide, mais je n'ai jamais eu besoin de me soucier de savoir si c'était le cas, car la formulation ci-dessus a toujours été suffisamment rapide pour mes besoins.

Pour moi, le message à retenir est d'utiliser la formulation la plus claire pour vous. Optimisez seulement si nécessaire. Personnellement, je trouve la formulation ci-dessus la plus claire, c'est pourquoi je l'utilise. Mais si la formulation inverse est plus claire pour vous, allez-y.

Answer 2

Une autre variation. Ainsi, vous obtenez lisibilité et bonnes performances:

NSMutableIndexSet *discardedItems = [NSMutableIndexSet indexSet];
SomeObjectClass *item;
NSUInteger index = 0;

for (item in originalArrayOfItems) {
    if ([item shouldBeDiscarded])
        [discardedItems addIndex:index];
    index++;
}

[originalArrayOfItems removeObjectsAtIndexes:discardedItems];

Answer 3

C'est un problème très simple. Il suffit d'itérer en arrière :

for (NSInteger i = array.count - 1; i >= 0; i--) {
   ElementType* element = array[i];
   if ([element shouldBeRemoved]) {
       [array removeObjectAtIndex:i];
   }
}

C'est un motif très courant.

Answer 4

Certaines des autres réponses pourraient avoir de mauvaises performances sur des tableaux très grands, car des méthodes comme removeObject: et removeObjectsInArray: impliquent une recherche linéaire du récepteur, ce qui est une perte de temps car vous savez déjà où se trouve l'objet. De plus, tout appel à removeObjectAtIndex: devra copier les valeurs de l'index à la fin du tableau un emplacement à la fois.

Plus efficace serait ce qui suit :

NSMutableArray *array = ...
NSMutableArray *itemsToKeep = [NSMutableArray arrayWithCapacity:[array count]];
for (id object in array) {
    if (! shouldRemove(object)) {
        [itemsToKeep addObject:object];
    }
}
[array setArray:itemsToKeep];

Comme nous avons défini la capacité de itemsToKeep, nous ne perdons pas de temps à copier des valeurs lors d'un redimensionnement. Nous ne modifions pas le tableau en place, donc nous sommes libres d'utiliser la numération rapide. Utiliser setArray: pour remplacer le contenu de array par itemsToKeep sera efficace. Selon votre code, vous pourriez même remplacer la dernière ligne par :

[array release];
array = [itemsToKeep retain];

Il n'est donc même pas nécessaire de copier les valeurs, il suffit de remplacer un pointeur.

Answer 5

Vous pouvez utiliser NSPredicate pour supprimer des éléments de votre tableau mutable. Cela ne nécessite pas de boucles for.

Par exemple, si vous avez un NSMutableArray de noms, vous pouvez créer un prédicat comme celui-ci:

NSPredicate *caseInsensitiveBNames = 
[NSPredicate predicateWithFormat:@"SELF beginswith[c] 'b'"];

La ligne suivante ne laissera dans votre tableau que les noms commençant par b.

[namesArray filterUsingPredicate:caseInsensitiveBNames];

Si vous avez du mal à créer les prédicats dont vous avez besoin, utilisez ce lien du développeur Apple.