区间求并集--python

原创已于 2023-06-21 11:07:57 修改 · 859 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数学建模 #python #算法

于 2023-06-21 10:45:22 首次发布

该文章介绍了一种方法，通过对区间列表排序，然后比较新区间与结果区间最后一个子区间的边界，来合并有交集的区间。当新区间开始点大于结果区间的结束点时，新区间作为一个独立区间添加；否则，更新结果区间的结束点为两者最大值。修改条件‘b[-1][1]>=begin’控制是否严格合并。

思想：先对区间list进行排序。每次加新区间时，对比【新区间开头和结尾】和【当前结果区间中最后一个子区间的结尾】，根据大小关系判断如何更新结果区间。如果开头大于结果区间结尾，也就是新区间是独立的一个区间，则直接将新区间加在结果区间最后。如果开头小于结果区间结尾，也就是说有两者有交集，则用最大的结尾更新结果区间的最后一个结尾。

因为之前用了排序，所以只考虑结果区间的最后一个子区间和新区间关系就行。

“b[-1][1] >= begin”这里按需求改吧。原来文章里写的是“b[-1][1] >= begin-1”，这会让（13，14）和（15，16）合成（13，16）。改成“b[-1][1] >= begin”后结果是（13，14）和（15，16）不合成，13，15）和（15，16）合成（13，16）.如果想开区间运算就“b[-1][1] > begin”

a = [(2, 4), (1,6), (11, 15), (14, 20), (23, 39)]
b = []
for begin,end in sorted(a):
        if b and b[-1][1] >= begin:
                b[-1][1] = max(b[-1][1], end)
        else:
                b.append([begin, end])
print(b)

参考：python - Union of multiple ranges - Stack Overflow