這個問的其實是數據結構中的二叉樹，查找一個普通二叉樹中兩個節點最近的公共祖先問題
假設兩個視圖為UIViewA、UIViewC，其中UIViewA繼承于UIViewB，UIViewB繼承于UIViewD，UIViewC也繼承于UIViewD；即A->B->D，C->D
方法一：
- (void)viewDidLoad {
[super viewDidLoad];
Class commonClass1 = [self commonClass1:[ViewA class] andClass:[ViewC class]];
NSLog(@"%@",commonClass1);
// 輸出：2018-03-22 17:36:01.868966+0800 兩個UIView的最近公共父類[84288:2458900] ViewD
}
// 獲取所有父類
- (NSArray *)superClasses:(Class)class {
if (class == nil) {
return @[];
}
NSMutableArray *result = [NSMutableArray array];
while (class != nil) {
[result addObject:class];
class = [class superclass];
}
return [result copy];
}
- (Class)commonClass1:(Class)classA andClass:(Class)classB {
NSArray *arr1 = [self superClasses:classA];
NSArray *arr2 = [self superClasses:classB];
for (NSUInteger i = 0; i < arr1.count; ++i) {
Class targetClass = arr1[i];
for (NSUInteger j = 0; j < arr2.count; ++j) {
if (targetClass == arr2[j]) {
return targetClass;
}
}
}
return nil;
}
方法一明顯的是兩層for循環，時間復雜度為O(N^2)
// 一個改進的辦法:我們將一個路徑中的所有點先放進NSSet中.因為NSSet的內部實現是一個hash表，所以查詢元素的時間的復雜度變成O(1),我們一共有N個節點，所以總時間復雜度優化到了O(N)
- (Class)commonClass2:(Class)classA andClass:(Class)classB{
NSArray *arr1 = [self superClasses:classA];
NSArray *arr2 = [self superClasses:classB];
NSSet *set = [NSSet setWithArray:arr2];
for (NSUInteger i =0; i<arr1.count; ++i) {
Class targetClass = arr1[i];
if ([set containsObject:targetClass]) {
return targetClass;
}
}
return nil;
}
代碼效果如下：
代碼
