iOS 面試題（一）：尋找最近公共 View --轉自唐巧

題目：找出兩個 UIView 的最近的公共 View，如果不存在，則輸出 nil 。 

分析：這其實是數據結構里面的找最近公共祖先的問題。

一個 UIViewController 中的所有 view 之間的關系其實可以看成一顆樹，UIViewController 的 view 變量是這顆樹的根節點，其它的 view 都是根節點的直接或間接子節點。

所以我們可以通過 view 的 superview 屬性，一直找到根節點。需要注意的是，在代碼中，我們還需要考慮各種非法輸入，如果輸入了 nil，則也需要處理，避免異常。以下是找到指定 view 到根 view 的路徑代碼：

+ (NSArray *)superViews:(UIView *)view {
    if (view == nil) {
        return @[];
    }
    NSMutableArray *result = [NSMutableArray array];
    while (view != nil) {
        [result addObject:view];
        view = view.superview;
    }
    return [result copy];
}

 

然后對於兩個 view A 和 view B，我們可以得到兩個路徑，而本題中我們要找的是這里面最近的一個公共節點。

一個簡單直接的辦法：拿第一個路徑中的所有節點，去第二個節點中查找。假設路徑的平均長度是 N，因為每個節點都要找 N 次，一共有 N 個節點，所以這個辦法的時間復雜度是 O（N^2）。

+ (UIView *)commonView_1:(UIView *)viewA andView:(UIView *)viewB {
    NSArray *arr1 = [self superViews:viewA];
    NSArray *arr2 = [self superViews:viewB];
    for (NSUInteger i = 0; i < arr1.count; ++i) {
        UIView *targetView = arr1[i];
        for (NSUInteger j = 0; j < arr2.count; ++j) {
            if (targetView == arr2[j]) {
                return targetView;
            }
        }
    }
    return nil;
}

 

一個改進的辦法：我們將一個路徑中的所有點先放進 NSSet 中。因為 NSSet 的內部實現是一個 hash 表，所以查找元素的時間復雜度變成了 O（1），我們一共有 N 個節點，所以總時間復雜度優化到了 O（N）。

+ (UIView *)commonView_2:(UIView *)viewA andView:(UIView *)viewB {
    NSArray *arr1 = [self superViews:viewA];
    NSArray *arr2 = [self superViews:viewB];
    NSSet *set = [NSSet setWithArray:arr2];
    for (NSUInteger i = 0; i < arr1.count; ++i) {
        UIView *targetView = arr1[i];
        if ([set containsObject:targetView]) {
            return targetView;
        }
    }
    return nil;
}

 

除了使用 NSSet 外，我們還可以使用類似歸並排序的思想，用兩個「指針」，分別指向兩個路徑的根節點，然后從根節點開始，找第一個不同的節點，第一個不同節點的上一個公共節點，就是我們的答案。代碼如下：

/* O(N) Solution */
+ (UIView *)commonView_3:(UIView *)viewA andView:(UIView *)viewB {
    NSArray *arr1 = [self superViews:viewA];
    NSArray *arr2 = [self superViews:viewB];
    NSInteger p1 = arr1.count - 1;
    NSInteger p2 = arr2.count - 1;
    UIView *answer = nil;
    while (p1 >= 0 && p2 >= 0) {
        if (arr1[p1] == arr2[p2]) {
            answer = arr1[p1];
        }
        p1--;
        p2--;
    }
    return answer;
}

 

我們還可以使用 UIView 的 isDescendant 方法來簡化我們的代碼，不過這樣寫的話，時間復雜度應該也是 O(N^2) 的。lexrus 提供了如下的 Swift 版本的代碼：

/// without flatMap
extension UIView {    
func commonSuperview(of view: UIView) -> UIView? {        
    if let s = superview {            
       if view.isDescendant(of: s) {                
                 return s
            } else {                
                 return s.commonSuperview(of: view)
            }
        }        
       return nil
    }
}

 

特別地，如果我們利用 Optinal 的 flatMap 方法，可以將上面的代碼簡化得更短，基本上算是一行代碼搞定。怎么樣，你學會了嗎？

extension UIView {
    func commonSuperview(of view: UIView) -> UIView? {
        return superview.flatMap { 
           view.isDescendant(of: $0) ? 
             $0 : $0.commonSuperview(of: view) 
        }
    }
}