AcWing刷题-区间合并

ofsurvival

于 2024-03-31 11:56:49 发布

阅读量378

点赞数 11

文章标签： python 算法

本文链接：https://blog.youkuaiyun.com/ofsurvival/article/details/137196106

版权

本文介绍了如何使用Python编写一个函数`merge`，对输入的区间列表进行合并，消除重叠部分，最后计算并输出所有非重叠区间的总长度。通过实例展示了区间合并算法在处理树状结构数据的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

校门外的树

在这里插入图片描述

区间合并：

from typing import List
def merge(intervals: List[List[int]]) -> List[List[int]]:
    # 按照第一个元素从小到大进行排序
    intervals.sort(key=lambda x: x[0])
    # 初始化一个新的数组
    new_list = list()
    for i in intervals:
        # 把第一个数组元素添加到新的数组
        if not new_list:
            new_list.append(i)
            continue
        # 如果有重合 就合并
        if new_list[-1][1] >= i[0]:
            x = max(new_list[-1][1], i[1])
            new_list[-1][1] = max(new_list[-1][1], i[1])
        # 如果没有重合 就直接添加到新的数组
        else:
            new_list.append(i)
    return new_list
L,M = map(int,input().split())
a = []
for _ in range(M):
    a.append(list(map(int,input().split())))
new_a = merge(a)
sum = 0
for elem in new_a:
    sum += elem[1]-elem[0]+1
print(L+1-sum)