有几家公司(华为、苹果、谷歌等)联合举行招聘会,发布了各家招聘的人数。
注意,招聘人数可能为0(HR闲闲没事做,招着玩)。
// 每家公司的招聘人数
// nums.size()为公司数
// nums[i]为该公司的招聘人数,招聘人数可以为0
const vector<int> &nums;
有几个求职者(王阳明、王重阳、王沪生等)一起参加招聘会,他们参加了笔试,填了入职意愿表(应聘哪几家公司)。
// 求职者的笔试成绩
// scores.size()为求职人数
const vector<int> &scores;
// 求职者的求职意愿表,preference.size() = scores.size()
// 1 <= preference[i].size() <= 3,即员工i至少想加入1家公司,至多想加入3家公司
const vector<vector<int>> &preference;
招聘会后,几家公司开始挑选人才,规则如下:
- 序号小的公司优先挑选;
- 每家公司只会挑选想加入该公司的人,即入职意愿表选了该公司才能录用;
- 择优录取,笔试成绩由高到低挑选;
- 招满为止,招聘人数为0视为招满;
- 宁缺勿滥。
最终要返回vector,内含两个元素:其一为没招满的公司个数,其二为没找到工作的求职者人数。
#include <algorithm>
#include <set>
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
vector<int> DepartmentRecruit(const vector<int> &nums,
const vector<int> &scores,
const vector<vector<int>> &preference)
{
candidateNum_ = scores.size();
vector<set<int>> prefVec(preference.size());
GetPrefVec(prefVec, preference);
vector<pair<int, int>> scorePairs;
for (int i = 0; i < scores.size(); i++) {
scorePairs.push_back({scores[i], i});
}
SortDesc(scorePairs);
GetAnswerOne(nums, prefVec, scorePairs);
answer[noJobCandidate_] = candidateNum_ - getJobNum_;
return answer;
}
private:
const int notFullCompany_ = 0;
const int noJobCandidate_ = 1;
int candidateNum_ = 0;
int getJobNum_ = 0;
vector<int> answer{ 0, 0 };
void SortDesc(vector<pair<int, int>> &scorePairs)
{
sort(scorePairs.begin(), scorePairs.end(),
[](pair<int, int> &a, pair<int, int> &b) {
if (a.first == b.first) {
return a.second < b.second;
}
return a.first > b.first;
}
);
return;
}
void GetPrefVec(vector<set<int>> &prefVec,
const vector<vector<int>> &preference)
{
for (int i = 0; i < preference.size(); i++) {
for (int j : preference[i]) {
prefVec[i].insert(j);
}
}
return;
}
void GetAnswerOne(const vector<int> &nums,
vector<set<int>> &prefVec,
vector<pair<int, int>> &scorePairs)
{
for (int i = 0; i < nums.size(); i++) {
int getNum = 0;
for (int j = 0; j < scorePairs.size(); j++) {
if (getNum == nums[i]) {
break;
}
auto &set = prefVec[scorePairs[j].second];
if (set.find(i) != set.end()) {
set.clear();
getJobNum_++;
getNum++;
}
}
if (getNum != nums[i]) {
answer[notFullCompany_]++;
}
}
return;
}
};