华为OD笔试机试 - 会议室占用时间 (python 2024年C卷D卷真题算法)

华为OD机试（C卷+D卷）2024真题目录(Java & c++ & python)

题目描述

现有若干个会议，所有会议共享一个会议室，用数组表示各个会议的开始时间和结束时间，格式为：

[[会议1开始时间, 会议1结束时间], [会议2开始时间, 会议2结束时间]]

请计算会议室占用时间段。

输入描述

[[会议1开始时间, 会议1结束时间], [会议2开始时间, 会议2结束时间]]

备注：

会议室个数范围：[1, 100]
会议室时间段：[1, 24]

输出描述

输出格式预输入一致,具体请看用例。

[[会议开始时间, 会议结束时间], [会议开始时间, 会议结束时间]]

用例1

输入：

[[1,4],[2,5],[7,9],[14,18]]

输出：

[[1,5],[7,9],[14,18]]

说明：

时/间段[1,4]和[2,5]重叠，合并为[1,5]

用例2

输入：

[[1,4],[4,5]]

输出：

[[1,5]]

说明：

时间段[1,4]和[4,5]连续

解题思路

区间合并问题
将所有区间开始起始位置升序，然后取出第一个区间作为基准值pre，从第二个区间cur开始遍历：

如果 cur.start <= pre.end，则说明两个区间有重叠，此时我们应该将两个区间合并，合并策略是将pre.end = max(pre.end, cur.end)，比如：

pre = [1, 4]，cur = [2, 5]，那么按此策略合并后，pre = [1, 5]

pre = [1, 100]，cur = [7, 9]，那么按此策略合并后，pre = [1, 100]

如果 cur.start > pre.end，则说明两个区间无交集，此时pre无法和后面任何区间合并（因为已经按照开始时间升序了，后面区间的开始时间肯定也大于pre.end），此时pre时间段就是一个独立的会议室占用时间，我们将它缓存记录下来，并将更新pre = cur，即将cur作为新的基准值和后面的区间比较

参考代码

def merge(meetings):
    # 按照会议的开始时间升序排序
    meetings.sort(key=lambda x: x[0])
 
    # 存储合并后的会议时间段
    merged_meetings = []
 
    # 当前会议时间段
    current_meeting = meetings[0]
    for i in range(1, len(meetings)):
        next_meeting = meetings[i]
 
        if next_meeting[0] <= current_meeting[1]:
            # 如果下一个会议的开始时间在当前会议的结束时间之前或相等，则合并
            current_meeting[1] = max(current_meeting[1], next_meeting[1])
        else:
            # 否则，不合并，将当前会议加入结果列表
            merged_meetings.append(current_meeting)
            current_meeting = next_meeting
 
    # 最后一个会议加入结果列表
    merged_meetings.append(current_meeting)
 
    return merged_meetings

# 示例输入输出处理
if __name__ == "__main__":
    n = int(input())
    meetings = [list(map(int, input().split())) for _ in range(n)]
    
    for start, end in merge(meetings):
        print(f"{start} {end}")