1056 Mice and Rice (25分)
参考
题目读了好久都没读明白。。。看了上方链接的解释才懂什么意思,感觉好绕。。。
第二行为0~n-1的老鼠重量,第三行为合并的次序序号,最后按0到n-1输出排行
排行为1 2 2 4形式
#include<iostream>
#include<vector>
#include<cmath>
#include<queue>
#include<algorithm>
using namespace std;
struct node {
int v, id, rank;
};
bool cmp(node a, node b) { return a.v > b.v; }
int main() {
int n, m, t;
cin >> n >> m;
vector<node> temp(n), a;
queue<node> q;
for (int i = 0; i < n; i++) {
cin >> temp[i].v;
temp[i].id = i;
}
for (int i = 0; i < n; i++) {
cin >> t;
q.push(temp[t]);
}
int cnt = 1, sum = q.size();
while (q.size() > 1) {
sum=ceil((float)sum / (float)m);//sum为下一波元素个数,也是当前可分的组数
int rank =sum + 1;//当前被淘汰的排行为组数+1
vector<node> p;
for (int i = 0; i < sum;i++) {
t = m;
while (q.size()&&t--) {
a.push_back(q.front());
q.pop();
}
sort(a.begin(), a.end(),cmp);
p.push_back(a[0]);
for (int j = 1; j < a.size(); j++) {
temp[a[j].id].rank = rank;
}
a.clear();
}
for (int i = 0; i < p.size(); i++) {q.push(p[i]);}
p.clear();
}
temp[q.front().id].rank = 1;
for (int i = 0; i < n; i++) {
if (i) { cout << ' '; }
cout << temp[i].rank;
}
return 0;
}