LOJ6130-冒泡排序，想法题

LOJ6130

题目描述

题解

首先我们发现如果对于一个数，他的前面大于他的数大于或等于$k$，那么它必定会往后移$k$位，那么对于其他的数，我们只需要按照从小到大的顺序依次插入剩余的空中，得到的便是最后的序列(感性证明)

代码

#include<bits/stdc++.h>
#define M 200100 
using namespace std;
int read(){
	int f=1,re=0;
	char ch;
	for(ch=getchar();!isdigit(ch)&&ch!='-';ch=getchar());
	if(ch=='-'){f=-1,ch=getchar();}
	for(;isdigit(ch);ch=getchar()) re=(re<<3)+(re<<1)+ch-'0';
	return re*f;
}
priority_queue<pair<int,int> >q;
int n,k,c[M],tr[M];
struct zb{int num,id;}a[M];
struct data{int x,y;}b[M];
int lowbit(int i){return i&(-i);}
void update(int x,int y){
	while(x<=n+5){
		tr[x]+=y;
		x+=lowbit(x);
	}
}
int query(int x){
	int ans=0;
	while(x>0){
		ans+=tr[x];
		x-=lowbit(x);
	}return ans;
}
bool cmp(const zb &a,const zb &b){
	if(a.num!=b.num) return a.num<b.num;
	else return a.id<b.id;
}
int main(){
	n=read(),k=read();
	for(int i=1;i<=n;i++) a[i].num=read(),a[i].id=i;
	sort(a+1,a+n+1,cmp);
	for(int i=1;i<=n;i++) b[a[i].id].x=i,b[a[i].id].y=a[i].num;
	for(int i=1;i<=n;i++){
		int w=i-query(b[i].x)-1;
		if(w>=k) c[i-k]=b[i].y;
		else q.push(make_pair(-b[i].x,i));
		update(b[i].x+1,1);
	}
	for(int i=1;i<=n;i++){
		if(!c[i]){
			int w=q.top().second;
			c[i]=b[w].y;
			q.pop();
		}
	}for(int i=1;i<=n;i++) printf("%d\n",c[i]);
	return 0;
}