最近项目需要对好友进行排序,由于中文的特殊性算法也改了很多次,
下面用一个Demo来说明一下的过程
首先创建一个用来测试数组,简单的User模型中只有一个name属性,遍历赋值。
NSArray *sortArray = @[@"泥嚎5",@"567",@"def",@"你好",@"泥嚎0",@"NIHAOMA1",@"nihaoma",@"abc",@"$*&^",@"123",@"啊"];
NSMutableArray *userModels = [NSMutableArray array];
for (NSString *name in sortArray) {
User *user = [User new];
user.name = name;
[userModels addObject:user];
}
首先需要找到合适的比较方法,开始只提取了汉字的拼音进行排序,但后来发现了有些汉字如“你好”和“泥嚎”拼音完全相同再进行汉字排序逻辑又会变得复杂。
但幸运的是在NSString类中下面三个方法加以封装就可以方便的比较中文,方法名带有localized描述的比较方法会根据当前语言来决定权重,比如中文环境下中文的顺序会高于英文字母
对数组排序使用sortedArrayUsingFunction:comparator:方法
- (NSArray *)sortedArrayUsingFunction:(NSInteger (*)(ObjectType, ObjectType, void * __nullable))comparator context:(nullable void *)context;
创建一个C函数作为上面方法的参数
NSInteger nickNameSort(id user1, id user2, void *context)
{
User *u1,*u2;
u1 = (User*)user1;
u2 = (User*)user2;
return [u1.name localizedCompare:u2.name];
}
调用方法,打印下排序后的结果:
NSArray *sortedArray = [userModels sortedArrayUsingFunction:nickNameSort context:NULL];
for (User *user in sortedArray) {
NSLog(@"%@",user.name);
}
结果是中文和英文顺序分别排好了,但英文并没有和中文实现混排,而且特殊字符也需要能独立处理。那么就得在上面思路基础上优化下方案了。
这里需要在正式排序前先按照首字符'A'~'Z'分组,然后对每个分组分别排序
- (void)sortArray:(NSArray *)toSortArray{
//将传入数组转换为可变数组
NSMutableArray *needSortArray = [NSMutableArray arrayWithArray:toSortArray];
//存储对应字母开头的所有数据的数组
NSMutableArray *classifiedArray = [[NSMutableArray alloc] init];
for(int i='A';i<='Z';i++){
NSMutableArray *rulesArray = [[NSMutableArray alloc] init];
NSString *indexString = [NSString stringWithFormat:@"%c",i];
for(int j = 0; j < needSortArray.count; j++){
User *model = [needSortArray objectAtIndex:j];
if([[self toPinyin: model.name] isEqualToString:indexString]){
//把model.name首字母相同的放到同一个数组里面
[rulesArray addObject:model];
[needSortArray removeObject:model];
j--;
}
}
if (rulesArray.count !=0) {
[classifiedArray addObject:rulesArray];
}
if (needSortArray.count == 0) {
break;
}
}
// 剩下的就是非字母开头数据,加在classifiedArray的后面
if (needSortArray.count !=0) {
[classifiedArray addObject:needSortArray];
}
//最后再分别对每个数组排序
NSMutableArray *sortCompleteArray = [NSMutableArray array];
for (NSArray *tempArray in classifiedArray) {
NSArray *sortedElement = [tempArray sortedArrayUsingFunction:nickNameSort context:NULL];
[sortCompleteArray addObject:sortedElement];
}
//sortCompleteArray就是最后排好序的二维数组了
}
上面用到的获得汉字拼音的方法
- (NSString *)toPinyin:(NSString *)str{
NSMutableString *ms = [[NSMutableString alloc]initWithString:str];
if (CFStringTransform((__bridge CFMutableStringRef)ms, 0,kCFStringTransformMandarinLatin, NO)) {
}
// 去除拼音的音调
if (CFStringTransform((__bridge CFMutableStringRef)ms, 0,kCFStringTransformStripDiacritics, NO)) {
if (str.length) {
NSString *bigStr = [ms uppercaseString];
NSString *cha = [bigStr substringToIndex:1];
return cha;
}
}
return str;
}
最后结果完美,全部按照字母顺序排序,分组中先中文后英文,所有其他字符也排好序放在最后面。