码迷,mamicode.com
首页 > 编程语言 > 详细

【bzoj5170】Fable(树状数组)

时间:2018-10-01 22:11:01      阅读:192      评论:0      收藏:0      [点我收藏+]

标签:display   pen   read   传送门   splay   class   http   bool   def   

  题目传送门: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]);
}
bzoj5170

 

【bzoj5170】Fable(树状数组)

标签:display   pen   read   传送门   splay   class   http   bool   def   

原文地址:https://www.cnblogs.com/quzhizhou/p/9735864.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!