1056. Mice and Rice (25)

感觉写的好繁琐。。。

#include<iostream>
#include<algorithm>
#include<vector>
#include<map>
#include<string>
#include<set>
#include<stack>
#include<queue>
using namespace std;
struct node{
int data;
int w;
int lev;
int r;
};
vector<node> num;
queue<int> q, p;
bool comp(node a, node b){
if(a.lev > b.lev) return true;
else return false;
}
bool comp1(node a, node b){
if(a.data < b.data) return true;
else return false;
}
int main(){
int n, k;
cin>>n>>k;
// int temp[n + 1];
// int w[n + 1];
// int lev[n + 1];
for(int i = 0; i < n; i++){
node a;
scanf("%d",&a.w);
a.data = i;
a.lev = 1;
num.push_back(a);
}
for(int i = 0; i < n; i++){
int ans;
scanf("%d",&ans);
q.push(ans);
}
while(q.size() != 1){
while(!q.empty()){
int count = 0;
int max = -1;
int maxi = 0;
while(count < k && !q.empty()){
int temp = q.front();
q.pop();
if(num[temp].w > max){
max = num[temp].w;
maxi = temp;
}
count++;
}
num[maxi].lev++;
p.push(maxi);
}
q = p;
while(!p.empty()){
p.pop();
}
}
sort(num.begin(),num.end(),comp);
int level = num[0].lev;
int j = 1;
int ran = 1;
for(int i = 0; i < n; i++){
if(num[i].lev == level){
num[i].r = ran;
j++;
}
else{
level = num[i].lev;
ran = j++;
num[i].r = ran;
}
}
sort(num.begin(),num.end(),comp1);
for(int i = 0; i < n; i++){
if(i != 0) printf(" ");
printf("%d",num[i].r);
}
return 0;
}