Objective-C zip 函数 用于二维数组重排

Objective-C_zip_Function.m

/** * 把一个二维数组做坐标转换。例；

 原数组：

 arry = [[a, b, c],

         [e, f, g]

        ]

 data = zip(array)

 变型后：

 data = [[a, e],

         [b, f],

         [c, g]

        ]

 注意：请确保原数组的每个子数组的元素数量是一致的。

 */



NSArray * zip(NSArray *arrays){

    NSInteger maxSize = 0;

    for (NSArray *array in arrays) {

        if (array.count > maxSize) {

            maxSize = array.count;

        }

    }

    NSMutableArray *result = [NSMutableArray arrayWithCapacity:maxSize];

    for (NSInteger index = 0; index < maxSize; index ++) {

        NSMutableArray *row = nil;

        if (result.count > index) {

            row = [result objectAtIndex:index];

        }else{

            row = [NSMutableArray arrayWithCapacity:arrays.count];

        }

        for (NSArray *originRow in arrays) {

            [row addObject:[originRow objectAtIndex:index]];

        }

        [result addObject:row];

    }

    return result;

}