Objective-C集合类型NSSet与NSMutableSet详解-育师

1. Objective-C集合类型概述

在Objective-C开发中，集合类型是处理对象组的基础工具。Foundation框架提供了三种主要集合类：NSArray、NSDictionary和NSSet。前两者广为人知，而NSSet及其可变版本NSMutableSet却常被开发者忽视。实际上，在处理需要快速查找和去重的场景时，NSSet系列展现出独特的性能优势。

NSSet代表一个无序的、唯一对象的集合。与NSArray不同，它不保证元素的存储顺序，但确保每个元素只出现一次。这种特性使其特别适合处理需要去重或快速成员检查的情况。NSMutableSet作为其可变子类，提供了动态修改集合内容的能力。

2. NSSet核心功能解析

2.1 创建与初始化

NSSet提供多种初始化方式，最常用的是setWithObjects:方法。这个方法接受可变数量的参数，以nil结尾：

NSSet *colorSet = [NSSet setWithObjects:@"Red", @"Green", @"Blue", nil];

需要注意的是，如果传入的多个对象相同，集合会自动去重。例如：

NSSet *numberSet = [NSSet setWithObjects:@1, @2, @2, @3, nil]; // 实际只包含1,2,3三个对象

2.2 关键操作方法

NSSet的核心方法包括：

count：返回集合中元素的数量
containsObject:：检查对象是否存在
allObjects：返回包含所有元素的数组
anyObject：随机返回一个元素（对算法优化很有用）

一个典型的使用场景是检查用户输入的有效性：

NSSet *validCommands = [NSSet setWithObjects:@"start", @"stop", @"pause", nil]; if (![validCommands containsObject:userInput]) { NSLog(@"Invalid command!"); }

2.3 集合运算

NSSet支持多种集合运算：

intersectsSet:：检查两个集合是否有交集
isSubsetOfSet:：判断是否为子集
isEqualToSet:：判断集合相等性

这些运算在处理权限系统时特别有用：

NSSet *userPermissions = [NSSet setWithObjects:@"read", @"write", nil]; NSSet *requiredPermissions = [NSSet setWithObjects:@"write", @"delete", nil]; if ([userPermissions intersectsSet:requiredPermissions]) { // 有部分权限 }

3. NSMutableSet扩展功能

3.1 动态修改操作

NSMutableSet继承了NSSet的所有特性，并添加了修改方法：

addObject:：添加单个对象
removeObject:：移除指定对象
addObjectsFromArray:：从数组批量添加
removeAllObjects：清空集合

一个实际应用是动态维护一个标签集合：

NSMutableSet *tags = [NSMutableSet set]; [tags addObject:@"iOS"]; [tags addObject:@"Objective-C"]; [tags addObject:@"Swift"]; [tags removeObject:@"Objective-C"];

3.2 批量修改方法

NSMutableSet还提供了集合级别的修改操作：

unionSet:：并集运算
minusSet:：差集运算
intersectSet:：交集运算

这些方法可以高效处理多个集合的关系：

NSMutableSet *setA = [NSMutableSet setWithObjects:@1, @2, @3, nil]; NSMutableSet *setB = [NSMutableSet setWithObjects:@3, @4, @5, nil]; [setA unionSet:setB]; // setA变为1,2,3,4,5 [setA intersectSet:setB]; // setA变为3

4. 性能特点与最佳实践

4.1 时间复杂度分析

NSSet使用哈希表实现，关键操作的时间复杂度为：

查找：O(1)
插入：O(1)
删除：O(1)

相比之下，NSArray的查找操作是O(n)。当需要频繁检查元素是否存在时，NSSet的性能优势明显。

4.2 对象要求

存储在NSSet中的对象必须：

实现isEqual:和hash方法
遵守规则：如果isEqual:返回YES，hash必须相同
在集合生命周期内保持hash值不变

4.3 使用场景建议

优先考虑NSSet的场景：

需要快速成员检查
需要自动去重
不关心元素顺序
经常进行集合运算

5. 常见问题与解决方案

5.1 自定义对象存储问题

当存储自定义类对象时，必须正确实现hash和isEqual:：

@interface Person : NSObject @property (nonatomic, copy) NSString *name; @property (nonatomic) NSInteger age; @end @implementation Person - (BOOL)isEqual:(id)object { if (self == object) return YES; if (![object isKindOfClass:[Person class]]) return NO; Person *other = (Person *)object; return [self.name isEqualToString:other.name] && self.age == other.age; } - (NSUInteger)hash { return self.name.hash ^ self.age; } @end

5.2 内存管理注意事项

在MRC环境下，集合会对对象进行retain/release操作。ARC环境下，强引用集合会导致循环引用风险：

__weak typeof(self) weakSelf = self; [self.completionHandlers addObject:^{ [weakSelf doSomething]; }];

5.3 枚举技巧

NSSet支持多种枚举方式：

// 快速枚举 for (id obj in set) { NSLog(@"%@", obj); } // Block枚举 [set enumerateObjectsUsingBlock:^(id obj, BOOL *stop) { if ([obj isEqual:target]) { *stop = YES; } }];

6. 进阶应用场景

6.1 替代数组去重

使用NSSet可以简洁地实现数组去重：

NSArray *duplicates = @[@1, @2, @2, @3, @1]; NSArray *unique = [[NSSet setWithArray:duplicates] allObjects];

6.2 高效缓存实现

基于NSMutableSet可以实现简单的缓存系统：

@interface ObjectCache : NSObject @property (nonatomic, strong) NSMutableSet *cache; @end @implementation ObjectCache - (BOOL)containsObject:(id)object { return [self.cache containsObject:object]; } - (void)addObject:(id)object { [self.cache addObject:object]; } - (void)removeExpiredObjects { NSMutableSet *expired = [NSMutableSet set]; for (id obj in self.cache) { if ([obj isExpired]) { [expired addObject:obj]; } } [self.cache minusSet:expired]; } @end

6.3 与NSCountedSet配合使用

Foundation还提供了NSCountedSet，可以记录元素的重复次数：

NSCountedSet *countedSet = [NSCountedSet set]; [countedSet addObject:@"Apple"]; [countedSet addObject:@"Apple"]; NSLog(@"Count: %lu", [countedSet countForObject:@"Apple"]); // 输出2

7. 调试与性能优化

7.1 调试输出技巧

NSSet的description方法默认输出无序内容。为方便调试，可以排序后输出：

NSLog(@"Sorted: %@", [[set allObjects] sortedArrayUsingSelector:@selector(compare:)]);

7.2 性能测试对比

通过简单测试可以直观看到性能差异：

NSMutableArray *array = [NSMutableArray array]; NSMutableSet *set = [NSMutableSet set]; NSUInteger count = 100000; // 插入测试 CFTimeInterval start = CACurrentMediaTime(); for (NSUInteger i = 0; i < count; i++) { [array addObject:@(i)]; } NSLog(@"Array insert: %f", CACurrentMediaTime() - start); start = CACurrentMediaTime(); for (NSUInteger i = 0; i < count; i++) { [set addObject:@(i)]; } NSLog(@"Set insert: %f", CACurrentMediaTime() - start); // 查找测试 start = CACurrentMediaTime(); [array containsObject:@(count-1)]; NSLog(@"Array search: %f", CACurrentMediaTime() - start); start = CACurrentMediaTime(); [set containsObject:@(count-1)]; NSLog(@"Set search: %f", CACurrentMediaTime() - start);

7.3 内存占用优化

对于大量小对象的存储，可以考虑使用NSHashTable的weak模式：

NSHashTable *weakTable = [NSHashTable weakObjectsHashTable]; [weakTable addObject:someObject];

8. 与Swift的互操作性

在Swift和Objective-C混编时，NSSet会自动桥接为Swift的Set：

// Objective-C NSSet *objcSet = [NSSet setWithObjects:@"A", @"B", nil]; // Swift中使用 let swiftSet = objcSet as! Set<String> swiftSet.contains("A") // true

反向操作也同样简单：

let swiftSet: Set<Int> = [1, 2, 3] let objcSet = swiftSet as NSSet

9. 实际项目应用案例

9.1 社交应用的好友关系

@interface User : NSObject @property (nonatomic, strong) NSMutableSet *friends; - (void)addFriend:(User *)user; - (void)removeFriend:(User *)user; - (BOOL)isFriendWith:(User *)user; @end @implementation User - (instancetype)init { if (self = [super init]) { _friends = [NSMutableSet set]; } return self; } - (void)addFriend:(User *)user { [self.friends addObject:user]; [user.friends addObject:self]; // 双向关系 } - (BOOL)isFriendWith:(User *)user { return [self.friends containsObject:user]; } @end

9.2 电商平台的商品分类

@interface Product : NSObject @property (nonatomic, strong) NSSet *categories; @end @interface CategoryManager : NSObject @property (nonatomic, strong) NSMutableSet *allCategories; - (void)addProducts:(NSSet *)products toCategory:(NSString *)category; @end

9.3 游戏中的碰撞检测

@interface GameEngine : NSObject @property (nonatomic, strong) NSMutableSet *collidableObjects; - (void)checkCollisions; @end @implementation GameEngine - (void)checkCollisions { for (GameObject *obj1 in self.collidableObjects) { for (GameObject *obj2 in self.collidableObjects) { if (obj1 != obj2 && [obj1 collidesWith:obj2]) { [obj1 handleCollisionWith:obj2]; } } } } @end

10. 最佳实践总结

在需要频繁检查元素是否存在时，优先选择NSSet而不是NSArray
自定义对象作为集合元素时，必须正确实现hash和isEqual:方法
注意NSMutableSet不是线程安全的，多线程环境需要额外同步
考虑使用NSCountedSet当需要统计元素出现次数时
大型集合操作考虑使用并行枚举提高性能
注意NSSet对nil值的处理：addObject:会抛出异常
使用unionSet:/minusSet:等批量操作比循环调用addObject:更高效

Objective-C集合类型NSSet与NSMutableSet详解