Python programmers interview book --- problem-solving algorithm Summary: Chapter 4 Array 4.23 How to find the intersection of two ordered sets

# -*- coding: utf-8 -*-

'''
Python程序员面试算法宝典---解题总结: 第4章 数组 4.23 如何求两个有序集合的交集

题目:
有两个有序的集合,集合中的每个元素都是一段范围,求其交集,
例如集合{[4, 8], [9, 13]}和{[6, 12]}的交集为
{[6, 8], [9, 12]}。

分析:
举例:
假设一个集合是[b1, e1],另一个集合是[b2, e2]
两者的交集如下:
b = max{b1, b2}
e = min{e1, e2}
如果b <= e,则有交集为[max{b1, b2}, min{e1, e2}];
否则,没有交集


关键:

参考:
Python程序员面试算法宝典
'''

def getIntersection(arr1, arr2):
    results = []
    if not arr1 or not arr2:
        return results
    len1 = len(arr1)
    len2 = len(arr2)
    i = 0
    j = 0
    while i < len1 and j < len2:
        b1 = arr1[i][0]
        e1 = arr1[i][1]
        b2 = arr2[j][0]
        e2 = arr2[j][1]
        b = max(b1, b2)
        e = min(e1, e2)
        # 如果当前元素有交集
        if b <= e:
            result = [b, e]
            results.append(result)
            # 接下来需要让较小的集合往前走
            if e1 < e2:
                i += 1
            else:
                j += 1
        else:
            # 如果第一个数组中的最小值大于第二个数组中最大值
            if b1 > e2:
                j += 1
            else:
                i += 1
    return results


def process():
    arr1 = [[4, 8], [9, 13]]
    arr2 = [[6, 12], [13, 15]]
    results = getIntersection(arr1, arr2)
    print results


if __name__ == "__main__":
    process()

 

Guess you like

Origin blog.csdn.net/qingyuanluofeng/article/details/93664501