pku2823 Sliding Window

离散化+树状数组找第k小元素

SBT又超时了,真搞不明白..

#include <iostream>
#include <algorithm>
using namespace std;

#define MAXN 1000001

int a[MAXN],c[MAXN],mx[MAXN],mxcnt,n,m;

struct Node{
    int num,tag;
}nod[MAXN];

class CP{
public:
    int operator()(Node a,Node b){
        return a.num<b.num;
    }
};

void prep(){
    mxcnt=0;
    memset(c,0,sizeof(c));
    int i;
    for(i=1;i<=n;i++){
        scanf("%d",&nod[i].num);
        nod[i].tag=i;
    }
    sort(nod+1,nod+n+1,CP());
    for(i=1;i<=n;i++)
        a[nod[i].tag]=i;
}

inline int lowbit(int x){
    return x&(-x);
}

void insert(int x,int t){
    while(x<=n){
        c[x]+=t;
        x+=lowbit(x);
    }
}

int findk(int k){
    int cnt=0,ans=0;
    for(int i=20;i>=0;i--){//i from log2(n) to 0
        ans+=(1<<i);
        if(ans>=n || cnt+c[ans]>=k)ans-=(1<<i);
        else cnt+=c[ans];
    }
    return ans+1;
}

int main(){
    int i;
    while(scanf("%d%d",&n,&m)!=EOF){
        prep();
    
        for(i=1;i<=m;i++)
            insert(a[i],1);
        printf("%d",nod[findk(1)].num);
        mx[mxcnt++]=nod[findk(m)].num;

        for(i=1;i<=n-m;i++){
            insert(a[i],-1);
            insert(a[i+m],1);
            printf(" %d",nod[findk(1)].num);
            mx[mxcnt++]=nod[findk(m)].num;
        }
        printf("\n");
        printf("%d",mx[0]);
        for(i=1;i<mxcnt;i++)
            printf(" %d",mx[i]);
        printf("\n");
    }
    return 0;
}