从带有子对象的数组中获取唯一对象的算法？

Question

我有一个数据类：

@interface Car : NSObject
{
  Engine *_engine;
}
//...

使用类：

@interface Engine : NSObject
{
  NSInteger _power;
  NSString *_name;
}
//...

和一个数组：

NSArray *myArray = [ // a bunch of Car objects with Engines

如何创建一个包含基于引擎“功率”的所有唯一项目的新数组？

例如：

0 带发动机的汽车（功率 = 20，名称 =“发动机一”）

1 带发动机的汽车（功率 = 45，名称 =“发动机二”）

2 带发动机的汽车（功率 = 20，名称 =“发动机三”） ）

最终数组将是汽车对象 0 和 1

两个数据类都有访问器，例如：

- (NSInteger)power;

Answer 1

NSArray *cars = ...;
NSMutableDictionary *carsByUniqueEnginePowers = [NSMutableDictionary dictionary];
for (Car *car in cars) {
    NSNumber *enginePower = [NSNumber numberWithInteger:car.engine.power];
    [carsByUniqueEnginePowers setObject:car forKey:enginePower];
}
NSArray *carsWithUniqueEnginePowers = [carsByUniqueEnginePowers allValues];

将根据发动机功率选择最后一个项目。

要获得第一个，请

[carsByUniqueEnginePowers setObject:car forKey:enginePower];

将此行替换为：

if (![carsByUniqueEnginePowers objectForKey:power]) {
    [carsByUniqueEnginePowers setObject:car forKey:enginePower];
}

Answer 2

枚举数组，构建一个索引列表，这些索引的对象具有以前未见过的 power 值。

创建一个集合来保存已经看到的power； -[NSArrayindexsOfObjectsPassingTest:] 处理构建索引列表。然后向原始数组询问其 objectsAtIndexes:。

NSMutableSet * powerSet = [NSMutableSet set];
NSIndexSet * indexes;
indexes = [carArr indexesOfObjectsPassingTest:^BOOL(id car, NSUInteger idx, BOOL *stop) {
                  // valueForKey: will return an NSNumber instance for scalar values.        
                  NSNumber * powerNum = [[car engine] valueForKey:@"power"];
                  // This power is already in the set; pass over this car.
                  if( [powerSet containsObject:powerNum] ){
                      return NO;
                  }
                  // Add the new power to the set and include this car in the index set.        
                  else {
                      [powerSet addObject:powerNum];
                      return YES;
                  }
              }];
NSArray * uniqueEnginePowerCars = [carArr objectsAtIndexes:indexes];