【模拟消息队列】Python 实现

import heapq

def simulate_message_queue():
    # 读取输入
    message_line = input().strip().split()
    consumer_line = input().strip().split()
    
    # 解析发布者的消息
    messages = []
    for i in range(0, len(message_line), 2):
        time = int(message_line[i])
        content = int(message_line[i + 1])
        messages.append((time, content))
    
    # 解析消费者的订阅和取消订阅
    consumers = []
    for i in range(0, len(consumer_line), 2):
        subscribe_time = int(consumer_line[i])
        unsubscribe_time = int(consumer_line[i + 1])
        consumers.append((subscribe_time, unsubscribe_time, i // 2))  # 消费者编号为i//2
    
    # 收集所有事件
    events = []
    for time, content in messages:
        events.append((time, 'message', content))
    for sub_time, unsub_time, consumer_id in consumers:
        events.append((sub_time, 'subscribe', consumer_id))
        events.append((unsub_time, 'unsubscribe', consumer_id))
    
    # 按时间排序事件,同一时刻的事件按订阅、取消订阅、消息的顺序处理
    events.sort(key=lambda x: (x[0], 0 if x[1] == 'subscribe' else (1 if x[1] == 'unsubscribe' else 2)))
    
    # 当前订阅的消费者,使用最小堆模拟最大优先级队列(优先级是负数)
    active_consumers = []
    consumer_messages = [[] for _ in range(len(consumers))]
    
    for event in events:
        time, event_type, data = event
        if event_type == 'subscribe':
            consumer_id = data
            # 使用负数模拟最大堆,因为消费者按优先级升序排列
            heapq.heappush(active_consumers, (-consumer_id, consumer_id))
        elif event_type == 'unsubscribe':
            consumer_id = data
            # 需要从堆中移除该消费者,这里简化处理,假设取消订阅时消费者一定在堆顶
            if active_consumers and active_consumers[0][1] == consumer_id:
                heapq.heappop(active_consumers)
            else:
                # 如果消费者不在堆顶,需要遍历整个堆来移除
                temp = []
                found = False
                while active_consumers:
                    item = heapq.heappop(active_consumers)
                    if item[1] == consumer_id:
                        found = True
                        break
                    temp.append(item)
                if not found:
                    for item in temp:
                        heapq.heappush(active_consumers, item)
        elif event_type == 'message':
            if active_consumers:
                # 获取优先级最高的消费者
                _, consumer_id = active_consumers[0]
                consumer_messages[consumer_id].append(data)
    
    # 输出结果
    for messages in consumer_messages:
        if messages:
            print(' '.join(map(str, messages)))
        else:
            print(-1)

simulate_message_queue()


相关文章

什么是队列?(Python队列)

前言上篇文章介绍了 Python 中的多线程。今天来介绍下编程中常会用到的一个数据结构 - 队列。不知道大家是否还记得什么是数据结构呢?在很早很早以前,Python小课堂的初期,讲了许多 Python...

Python高级编程之消息队列(Queue)与进程池(Pool)实例详解

Queue消息队列1.创建import multiprocessing queue = multiprocessing.Queue(队列长度) 2.方法3.进程通信嗨喽:正在学习python的小伙伴或...

python并发编程-队列介绍

进程彼此之间互相隔离,要实现进程间通信(IPC),multiprocessing模块支持两种形式:队列和管道(不推荐使用),这两种方式都是使用消息传递的 创建队列的类(底层就是以管道和锁定的方式实现)...

浅析 Python 中的队列类

Python中的队列源代码位于 lib/queue.py 「链接」中该queue模块实现了多生产者、多消费者队列。当必须在多个线程之间安全地交换信息时,queue模块特别有用。该Queue模块中的类实...

Python | 数据结构 - 队列

队列的 Python 代码实现队列是一种先进先出的数据类型。应用场景:我们的计算机实验室有 30 台计算机与一台打印机联网。当学生想要打印时,他们的打印任务与正在等待的所有其他打印任务 “一致”。第一...