题目描述
公司某部门软件教导团正在组织新员工每日打卡学习活动,他们开展这项学习活动已经一个月了,所以想统计下这个月优秀的打卡员工。每个员工会对应一个id,每天的打卡记录记录当天打卡员工的id集合,一共30天。
请你实现代码帮助统计出打卡次数top5的员工。加入打卡次数相同,将较早参与打卡的员工排在前面,如果开始参与打卡的时间还是一样,将id较小的员工排在前面。
注:不考虑并列的情况,按规则返回前5名员工的id即可,如果当月打卡的员工少于5个,按规则排序返回所有有打卡记录的员工id。
输入描述
第一行输入为新员工数量N,表示新员工编号id为0到N-1,N的范围为[1,100]
第二行输入为30个整数,表示每天打卡的员工数量,每天至少有1名员工打卡。
之后30行为每天打卡的员工id集合,id不会重复。
11
4 4 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 2 2
0 1 7 10
0 1 6 10
10
10
10
10
10
10
10
10
10
10
10
10
10
10
10
10
10
10
10
10
10
10
10
10
10
10
6 10
7 10
输出描述
按顺序输出打卡top5员工的id,用空格隔开。
10 0 1 7 6
代码
#include <bits/stdc++.h>
using namespace std;
int main() {
int n;
cin >> n;
vector<int> count(30);
for (int day = 0; day < 30; day++) {
cin >> count[day];//第day天打卡的员工数量
}
// key记录员工id, value记录员工打卡天数
map<int, int> freq;
// key记录员工id,value记录员工第一次打卡是哪天
map<int, int> first;
for (int day = 0; day < 30; day++) {
for (int i = 0; i < count[day]; i++) {
int id;
cin >> id;
freq[id]++;
//记录其第一次打卡的天数。
if (first.find(id) == first.end()) {
first[id] = day;
}
}
}
vector<int> ids;
for (const auto &p: freq) {
ids.emplace_back(p.first);
}
sort(ids.begin(), ids.end(), [&](int a, int b) {
// 打卡天数多的员工排在前面
if (freq[a] != freq[b]) {
return freq[a] > freq[b];
}
// 首次打卡早的员工排在前面
if (first[a] != first[b]) {
return first[a] < first[b];
}
// id小的员工排在前面
return a < b;
});
// 取打卡前5名
for (int i = 0; i < min((int) ids.size(), 5); i++) {
cout << ids[i] << " ";
}
return 0;
}