模拟消息队列 - 华为OD机试刷题记录
题目描述
让我们来模拟一个消息队列的运作,有一个发布者和若干消费者,发布者会在给定的时刻向消息队列发送消息,若此时消息队列有消费者订阅,这个消息会被发送到订阅的消费者中优先级最高(输入中消费者按优先级升序排列)的一个; 若此时没有订阅的消费者,该消息被消息队列丢弃。 消费者则会在给定的时刻订阅消息队列或取消订阅。当消息发送和订阅发生在同一时刻时,先处理订阅操作,即同一时刻订阅的消费者成为消息发送的候选。 当消息发送和取消订阅发生在同一时刻时,先处理取消订阅操作,即消息不会被发送到同一时刻取消订阅的消费者。
输入描述
输入为两行:
- 第一行为2N个正整数,代表发布者发送的N个消息的时刻和内容(为方便解折,消息内容也用正整数表示)。第一个数字是第一个消息的发送时刻,第二个数字是第一个消息的内容,以此类推。用例保证发送时刻不会重复,但注意消息并没有按照发送时刻排列。
- 第二行为2M个正整数,代表M个消费者订阅和取消订阅的时刻。第一个数字是第一个消费者订阅的时刻,第二个数字是第一个消费者取消订阅的时刻,以此类推。用例保证每个消费者的取消订阅时刻大于订阅时刻,消费者按优先级升序排列。
两行的数字都由空格分隔。N不超过100,M不超过10,每行的长度不超过1000字符
输出描述
- 输出为M行,依次为M个消费者收到的消息内容,消息内容按收到的顺序排列,且由空格分隔;若某个消费者没有收到任何消息,则对应的行输出-1
用例1
输入
2 22 1 11 4 44 5 55 3 33
1 7 2 3
输出
11 33 44 55
22
说明
说明 消息11在1时刻到达,此时只有第一个消费者订阅,消息发送给它;
消息22在2时刻到达,此时两个消费者都订阅了,消息发送给优先级最高的第二个消费者;
消息33在时刻3到达,此时只有第一个消费者订阅,消息发送给它;
余下的消息按规则也是发送给第一个消费者
用例2
输入
5 64 11 64 9 97
9 11 4 9
输出
97
64
说明
说明 消息64在5时刻到达,此时只有第二个消费者订阅,消息发送给它;
消息97在9时刻到达,此时只有第一消费者订阅(因为第二个消费者刚好在9时刻取消订阅),消息发送给它;
11时刻也到达了一个内容为64的消息,不过因为没有消费者订阅,消息被丢弃
参考题库
题解代码
#include<iostream>
#include<vector>
#include<string>
#include <utility>
#include <sstream>
#include<algorithm>
using namespace std;
// 通用 split 函数
vector<int> split(const string& str, const string& delimiter) {
vector<int> result;
size_t start = 0;
size_t end = str.find(delimiter);
while (end != string::npos) {
result.push_back(stoi(str.substr(start, end - start)));
start = end + delimiter.length();
end = str.find(delimiter, start);
}
// 添加最后一个部分
result.push_back(stoi(str.substr(start)));
return result;
}
void handle(vector<int> messageArr, vector<int> subArr) {
vector<pair<int, int>> messages;
for (int i = 0; i < messageArr.size(); i+=2) {
messages.push_back({messageArr[i], messageArr[i+1]});
}
// 消息按照时间进行排序
sort(messages.begin(), messages.end(), [](const pair<int,int> &a, const pair<int,int> &b){
return a.first < b.first;
});
vector<pair<int, int>> sub;
for (int i = 0; i < subArr.size(); i+=2) {
sub.push_back({subArr[i], subArr[i+1]});
}
vector<vector<int>> res(sub.size());
// 处理接收消息
for (int i = 0; i < messages.size(); i++) {
int sendTime = messages[i].first;
int content = messages[i].second;
// 按照优先级处理消息
for (int j = sub.size()-1; j >=0; j--) {
int subTime = sub[j].first;
int unsubTime = sub[j].second;
if (sendTime >= subTime && sendTime < unsubTime) {
res[j].push_back(content);
break;
}
}
}
// 处理输出
for (int i = 0; i < res.size(); i++) {
vector<int> tmp = res[i];
if (tmp.empty()) {
cout << "-1" << endl;
continue;
}
for (int j = 0; j < tmp.size(); j++) {
if (j != 0) {
cout << " ";
}
cout << tmp[j];
}
cout << endl;
}
}
int main() {
string message;
getline(cin , message);
string sub;
getline(cin, sub);
vector<int> messageArr = split(message, " ");
vector<int> subArr = split(sub, " ");
handle(messageArr, subArr);
return 0;
}
#华为OD##华为OD2025A卷##华为OD机考#