【bzoj5170】Fable(树状数组)
题目传送门:https://www.lydsy.com/JudgeOnline/problem.php?id=5170
我们会发现,经过一轮冒泡后,若a[i]的前面有比它大的数,就一定会有一个被丢到后面去,即a[i]会前移一位。而那些前面没有能被丢的数,它们是按照顺序从小到大排序好的。于是我们可以用树状数组处理出在他前面比他大的数个数>=k的数先插到前移k位的位置上,然后剩下的数从小到大插进空位。
代码:
#include<cstdio> #include<algorithm> #define ll long long #define maxn 200010 ll read() { ll x=0; char f=1,c=getchar(); for(;c<'0'||'9'<c;c=getchar())f=(c=='-'?0:f); for(;'0'<=c&&c<='9';c=getchar())x=x*10+c-'0'; return f?x:-x; } using namespace std; int bit[maxn]; struct data{ int x,id; }a[maxn]; int rk[maxn],f[maxn],ans[maxn]; int n,k; bool cmp(data a,data b){return a.x<b.x;} void add(int x,int k){for(;x<=n;x+=x&(-x))bit[x]+=k;} int getsum(int x){int sum=0; for(;x;x-=x&(-x))sum+=bit[x]; return sum;} int main() { n=read(); k=read(); for(int i=1;i<=n;i++) a[i].x=read(),a[i].id=i; sort(a+1,a+n+1,cmp); // for(int i=1;i<=n;i++) // printf("%d %d %d\n",i,a[i].x,a[i].id); rk[a[1].id]=1; for(int i=2;i<=n;i++) if(a[i].x==a[i-1].x)rk[a[i].id]=rk[a[i-1].id]; else rk[a[i].id]=i; for(int i=1;i<=n;i++){ add(rk[i],1); f[i]=getsum(n)-getsum(rk[i]); if(f[i]>k)ans[i-k]=a[rk[i]].x; } int tmp=1; for(int i=1;i<=n;i++) if(f[a[i].id]<=k){ while(tmp<=n&&ans[tmp])++tmp; ans[tmp]=a[i].x; } for(int i=1;i<=n;i++) printf("%d\n",ans[i]); // for(int i=1;i<=n;i++) // printf("%d %d %d\n",i,rk[i],f[i]); }