求两个有序集合的交集----【二维数组增加元素】【numpy】【Python】

题目描述

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

思路

循环遍历两个不同集合中的区间，两两作比较，只有一个区间是另一个区间的子集的时候，才能找到交集。当两个集合作比较的时候，只需要关注边界元素的值。

代码实现

import numpy as np

def Intersection(li1,li2):
    li3=np.empty([0,2],int)  #创建一个两列的二维数组，此时数组为空
    #print(li3)
    for i in range(len(li1)):
        for j in range(len(li2)):
            a=max(li1[i][0],li2[j][0])
            b=min(li1[i][1],li2[j][1])
            if a==li1[i][0]:
                f1=li1[i]
            if a==li2[j][0]:
                f1=li2[j]
            if b==li1[i][1]:
                f2=li1[i]
            if b==li2[j][1]:
                f2=li2[j]
            if f2[1]>=f1[0]:
                li3=np.append(li3,[[f1[0],f2[1]]],axis=0)   #向数组中增加元素，axis=0表示纵向增加，axis=1表示往横向增加  
    return li3
#测试
l1=[[4,8],[9,13]]
l2=[[6,12]]
print(Intersection(l1,l2))

输出