一個(gè)高效的條件篩選方法

背景:有一個(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];
}
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容