牛牛任职于一家演艺公司,这一天他率领着一共n个小伙伴去某知名剧组面试,在戏中,往往以戏份多少来定义角色的重要性。
现在这n个小伙伴对自己都有一个定你位,想要拿一个多少细分的角色。假设将戏份多少量化,以1~100来衡量,那么也就是说,每个人在心中都有一个指标,自己签订的合约角色戏份绝对不能少于这个指标。
在n个小伙伴都进入剧组面试之后,牛牛得到该剧组剩余m个角色的戏份值。牛牛想提前知道,在能够选角的人数最多的情况下,每个人的角色分配,请你输出一种可能的分配方案,
给定案例
1(测试案例数目)
3 6(分别为n和m)
33 66 99(n个人的预期指标)
3 6 9 30 60 90(m个剩余角色值)
输出 5 6 -1
两次排序,从最大戏份开始找
#include <iostream>
#include <vector>
#include <algorithm>
#include <functional>
using namespace std;
int main() {
int t;
cin >> t;
for (int i = 0; i < t; ++i) {
int n, m;
cin >> n >> m;
vector<vector<int> > parter;
vector<vector<int> > com;
for (int j = 0; j < n; ++j) {
int x;
cin >> x;
parter.push_back({ x,j + 1 });
}
for (int j = 0; j < m; ++j) {
int x;
cin >> x;
com.push_back({ x,j + 1 });
}
sort(parter.begin(), parter.end(), [](vector<int>& a, vector<int>& b) {
return a[0] > b[0];
});
sort(com.begin(), com.end(), [](vector<int>& a, vector<int>& b) {
return a[0] > b[0];
});
for (int i = 0,j = 0; i < parter.size(); ++i) {
if (com[j][0] >= parter[i][0]) {
parter[i][0] = com[j++][1];
}
else{
parter[i][0] = -1;
}
}
sort(parter.begin(), parter.end(), [](vector<int>& a, vector<int>& b) {
return a[1] < b[1];
});
for (vector<vector<int>>::iterator it = parter.begin(); it != parter.end(); ++it) {
cout << (*it)[0]<<" ";
}
cout << endl;
}
return 0;
}
class Compare {
public:
bool operator()(vector<int> &a, vector<int> &b) {
return a[0] > b[0];
}
};
sort(parter.begin(), parter.end(), Compare());
sort(com.begin(), com.end(), Compare());