首页 > 代码库 > 数组去重

数组去重

1、对数组中的元素去重复

例如:

 

1
NSArray *array = @[@"12-11", @"12-11", @"12-11", @"12-12", @"12-13", @"12-14"];

参考答案:

  • 第一种方法:开辟新的内存空间,然后判断是否存在,若不存在则添加到数组中,得到最终结果的顺序不发生变化。效率分析:时间复杂度为O ( n2):

 

 

1
2
3
4
5
6
7
8
9
10
NSMutableArray *resultArray = [[NSMutableArray alloc] initWithCapacity:array.count];
// 外层一个循环
for (NSString *item in array) {
   // 调用-containsObject:本质也是要循环去判断,因此本质上是双层遍历
   // 时间复杂度为O ( n^2 )而不是O (n)
    if (![resultArray containsObject:item]) {
      [resultArray addObject:item];
    }
}
NSLog(@"resultArray: %@", resultArray);

补充:原来集合操作可以通过valueForKeyPath来实现的,去重可以一行代码实现:

 

1
2
array = [array valueForKeyPath:@"@distinctUnionOfObjects.self"];
NSLog(@"%@", array);

但是返回的结果是无序的,与原来的顺序不同。大家可以阅读:Collection Operators

  • 第二种方法:利用NSDictionary去重,字典在设置key-value时,若已存在则更新值,若不存在则插入值,然后获取allValues。若不要求有序,则可以采用此种方法。若要求有序,还得进行排序。效率分析:只需要一个循环就可以完成放入字典,若不要求有序,时间复杂度为O(n)。若要求排序,则效率与排序算法有关:

 

 

1
2
3
4
5
6
NSMutableDictionary *resultDict = [[NSMutableDictionary alloc] initWithCapacity:array.count];
for (NSString *item in array) {
    [resultDict setObject:item forKey:item];
}
NSArray *resultArray = resultDict.allValues;
NSLog(@"%@", resultArray);

如果需要按照原来的升序排序,可以这样:

 

1
2
3
4
5
6
resultArray = [resultArray sortedArrayUsingComparator:^NSComparisonResult(id  _Nonnull obj1, id  _Nonnull obj2) {
  NSString *item1 = obj1;
  NSString *item2 = obj2;
  return [item1 compare:item2 options:NSLiteralSearch];
}];
NSLog(@"%@", resultArray);
  • 第三种方法:利用集合NSSet的特性(确定性、无序性、互异性),放入集合就自动去重了。但是它与字典拥有同样的无序性,所得结果顺序不再与原来一样。如果不要求有序,使用此方法与字典的效率应该是差不多的。效率分析:时间复杂度为O (n):

 

 

1
2
3
NSSet *set = [NSSet setWithArray:array];
NSArray *resultArray = [set allObjects];
NSLog(@"%@", resultArray);

如果要求有序,那就得排序,比如这里要升序排序:

 

1
2
3
4
5
6
resultArray = [resultArray sortedArrayUsingComparator:^NSComparisonResult(id  _Nonnull obj1, id  _Nonnull obj2) {
  NSString *item1 = obj1;
  NSString *item2 = obj2;
  return [item1 compare:item2 options:NSLiteralSearch];
}];
NSLog(@"%@", resultArray); 

 第四种方法

1
2
NSOrderedSet *set = [NSOrderedSet orderedSetWithArray:array];
NSLog(@"%@", set.array);

数组去重