Как я могу с легкостью сгладить трехмерный массив в Cocoa?

Скажем, у меня есть такой массив:

NSArray *threeDimensionalArray = @[
 @[     
     @[ @"Peter", @"Paul", @"Mary" ], @[ @"Joe", @"Jane" ]
  ],
 @[
     @[ @"Alice", @"Bob" ]
  ]
];

и я хочу, чтобы он стал:

@[ @"Peter", @"Paul", @"Mary", @"Joe", @"Jane", @"Alice", @"Bob" ]

Как я могу с легкостью создать этот сплющенный массив?

Ответы

Ответ 1

Оператор коллекции ключевых слов (KVC) @unionOfArrays выравнивает один уровень массивов, поэтому его применение дважды дает желаемый результат.

Операторы коллекции (кроме @count) нуждаются в ключевом пути к свойству коллекции, и поскольку наши объекты уже являются массивами (и, следовательно, наборами) сами по себе, путь ключа должен быть self.

Поэтому нам нужно дважды применить @unionOfArrays с ключом self, получив следующий вызов KVC для сглаживания трехмерного массива:

NSArray *flattenedArray = [threeDimensionalArray valueForKeyPath: @"@[email protected]"];

Ответ 2

Я понимаю, что этот поток немного стар, но мне нужно решение, где количество уровней в глубине не имеет большого значения. Следующие методы могут быть добавлены в категорию на NSArray. Я также включил тест для этих методов:

// This is the method that would be used from an outside class
- (NSArray *)flatten {
    NSArray *array = self;
    while (![array isFlattened]) {
        array = [array flattenOneLevel];
    }
    return [NSArray arrayWithArray:array];
}

- (NSArray *)flattenOneLevel {
    NSMutableArray *array = [NSMutableArray array];
    for (id object in self) {
        [object isKindOfClass:self.class] ? [array addObjectsFromArray:object] : [array addObject:object];
    }
    return array;
}

- (BOOL)isFlattened {
    BOOL flattened = YES;
    for (id object in self) {
        if ([object isKindOfClass:self.class]) {
            flattened = NO;
            break;
        }
    }
    return flattened;
}

В этом тесте для этих методов убедитесь, что он работает правильно:

it(@"should flatten an array", ^{
    NSArray *initialArray = @[@[@23, @354, @1, @[@7], @[@[@3]]], @[@[@890], @2, @[@[@6], @8]]];
    NSArray *expectedArray = @[@23, @354, @1, @7, @3, @890, @2, @6, @8];
    expect([initialArray flatten]).equal(expectedArray);
});

Ответ 3

Альтернативный ответ с использованием рекурсии, который займет больше памяти (в стеке), но для тех, кто любит рекурсию, проще читать:

- (NSArray *) flatten;
{
    NSMutableArray *flattedArray = [NSMutableArray new];

    for (id item in self) {
        if ([[item class] isSubclassOfClass:[NSArray class]]) {
            [flattedArray addObjectsFromArray:[item flatten]];
        } else {
            [flattedArray addObject:item];
        }
    }

    return flattedArray;
}

И продление тестов:

+ (void) unitTests;
{
    NSArray *flattenedArray;

    NSArray *initialArray1 = @[@[@23, @354, @1, @[@7], @[@[@3]]], @[@[@890], @2, @[@[@6], @8]]];
    NSArray *expectedArray1 = @[@23, @354, @1, @7, @3, @890, @2, @6, @8];
    flattenedArray = [initialArray1 flatten];
    SPASLogDetail(@"flattenedArray: %@", flattenedArray);
    AssertIf(![flattenedArray isEqualToArray:expectedArray1], @"Arrays are not equal");

    NSArray *initialArray2 = @[@[@23, @354, @1, [@[@7] mutableCopy], @[@[@3]]], @[[@[@890] mutableCopy], @2, @[@[@6], @8]]];
    NSArray *expectedArray2 = expectedArray1;
    flattenedArray = [initialArray2 flatten];
    SPASLogDetail(@"flattenedArray: %@", flattenedArray);
    AssertIf(![flattenedArray isEqualToArray:expectedArray2], @"Arrays are not equal");
}