背景:有一個(gè)元素為obj的數(shù)組,需要將其根據(jù)月份進(jìn)行篩選,由每個(gè)月的數(shù)據(jù)組成數(shù)組,再將其作為元素組成一個(gè)數(shù)組。
eg:
原數(shù)組:@[
@{month:2018-12,num:10},
@{month:2018-10,num:9},
@{month:2018-11,num:5},
@{month:2018-12,num:8},
@{month:2018-11,num:9}
]
處理后的數(shù)組: @[
@[ @{month:2018-12,num:10}, @{month:2018-12,num:8}],
@[ @{month:2018-10,num:9}],
@[@{month:2018-11,num:5}, @{month:2018-11,num:9}]
]
難點(diǎn):要在數(shù)組遍歷的同時(shí)分組,并將其組成新的數(shù)組。否則若先便利確定小分組個(gè)數(shù),就需要多次for循環(huán)。
解決辦法:如難點(diǎn)所說(shuō),設(shè)計(jì)代碼,使其遍歷的同時(shí)組成新的數(shù)組,這樣可以降低時(shí)間復(fù)雜度。代碼如下:
/**
將數(shù)據(jù)按月分組
@param dataArr 源數(shù)據(jù)
@return NSArray 處理后的數(shù)據(jù)
*/
- (NSArray *)sectionDataArrayWithArray:(NSArray *)dataArr {
NSMutableArray *tempArr = [[NSMutableArray alloc] init]; // 處理后的數(shù)組
// 先取所有的日期
NSMutableArray *yyyyArr = [[NSMutableArray alloc] init]; // 數(shù)組內(nèi)每個(gè)小分組
[dataArr enumerateObjectsUsingBlock:^(RecordModel *obj, NSUInteger idx, BOOL * _Nonnull stop) {
NSString *yyyymm = [NSDate dateStringWithString:obj.applydate inFormatString:DLDateFormatter_yyyyMMdd outFormatString:DLDateFormatter_yyyyMM];
__block NSInteger index = -1;
[yyyyArr enumerateObjectsUsingBlock:^(NSString *obj, NSUInteger idx, BOOL * _Nonnull stop) {
if ([obj isEqualToString:yyyymm]) {
index = idx;
*stop = YES;
}
}]; // 判斷 yyyyArr 是否包含了 yyyymm,若包含直接退出遍歷,index = idx (即yyyymm在yyyyArr中的位置),不包含,完成遍歷后 index 仍為 -1
if (index != -1) { // 包含,在 yyyyArr 中加入 當(dāng)前元素 yyyymm
[tempArr[index] addObject:obj];
} else { // 不包含,創(chuàng)建一個(gè)新的yyyyArr,將其加入 tempArr
[yyyyArr addObject:yyyymm];
[tempArr addObjectsFromArray:[@[obj] mutableCopy]];
}
}];
return [tempArr copy];
}