Acwing1542. 老鼠和大米

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;
}