https://www.acwing.com/problem/content/1544/
一道甲级大模拟题目
#include<bits/stdc++.h>
using namespace std;
const int N = 1e3+10;
int n,m;
int w[N];
vector<int> level[N],a,b;
map<int,int> hs;
int main()
{
cin>>n>>m;
for(int i=0;i<n;i++) cin>>w[i];
for(int i=0;i<n;i++)
{
int x; cin>>x;
a.push_back(w[x]);
}
int k=n,l=0;
while(k)
{
//cout<<"K: "<<k<<endl;
// for(int i=0;i<k;i++) cout<<a[i]<<' ';
// cout<<endl;
for(int i=0;i<k;i+=m)
{
//cout<<i<<' ';
int t=i;
for(int j=i;j<i+m&&j<k;j++) if(a[j]>a[t]) t=j;
for(int j=i;j<i+m&&j<k;j++)
if(j==t) b.push_back(a[j]);
else level[l].push_back(a[j]);
if(i+m>=k) break;
}
//cout<<"\n-------------------\n";
a=b;
b.clear();
l++;
k=a.size();
if(k==1)
{
level[l].push_back(a[0]);
break;
}
}
// for(int i=l;i>=0;i--)
// {
// cout<<i;
// for(int j=0;j<level[i].size();j++) cout<<' '<<level[i][j]<<' ';
// cout<<endl;
// }
int ll=1,rr=1;
for(int i=l;i>=0;i--)
{
for(int j=0;j<level[i].size();j++)
{
hs[level[i][j]]=ll;
rr++;
}
ll=rr;
}
for(int i=0;i<n;i++) cout<<hs[w[i]]<<' ';
cout<<endl;
return 0;
}