测试点1、2没过,不知道错哪了。看大家都是测试点4超时或者错误。以后改吧。。。
#include<bits/stdc++.h>
using namespace std;
struct stu {
int ge, gi, rank, id, gf;
vector<int> aply;
};
vector<stu> v;
vector<int> sch;
vector<vector<int>> ans;
bool cmp (stu a, stu b) {
if (a.gf != b.gf) return a.gf > b.gf;
return a.ge > b.ge;
}
int main() {
int n, m, k, rank = 1;
scanf ("%d %d %d", &n, &m, &k);
sch.resize(m); ans.resize(m);
for (int i = 0; i < m; i++) scanf ("%d", &sch[i]);
for (int i = 0; i < n; i++) {
stu temp;
temp.aply.resize(k);
scanf ("%d %d", &temp.ge, &temp.gi);
temp.gf = temp.ge + temp.gi;
for (int j = 0; j < k; j++) scanf ("%d", &temp.aply[j]);
temp.id = i;
v.push_back(temp);
}
sort (v.begin(), v.end(), cmp);
v[0].rank = 1;
for (int i = 1; i < v.size(); i++) {
if (v[i].gf == v[i - 1].gf && v[i].ge == v[i - 1].ge) {
v[i].rank = v[i - 1].rank;
}
else {
v[i].rank = i + 1;
}
}
for (int i = 0; i < v.size(); i++) {
for (int j = 0; j < k; j++) {
int now = v[i].aply[j];
if (sch[now] > 0) {
ans[now].push_back(v[i].id);
sch[now]--;
break;
}
else {
int fRank = v[ans[now][ans[now].size() - 1]].rank;
if (v[i].rank == fRank) {
ans[now].push_back(v[i].id);
break;
}
}
}
}
for (int i = 0; i < m; i++) {
if (ans[i].size() == 0) printf("\n");
else {
sort(ans[i].begin(), ans[i].end());
for (int j = 0; j < ans[i].size(); j++) {
if (j != 0) printf (" ");
printf ("%d", ans[i][j]);
}
printf ("\n");
}
}
}