Codechef August Challenge 2018 : Coordinate Compression

传送门

外边二分，里面拿线段树维护贪心就行了。

#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#define MN 110000
#define lp p<<1
#define rp p<<1|1
using namespace std;

vector<int> v[MN];
int t,n,s,a[MN],o[MN],z[MN<<];
bool cmp(int x,int y){return a[x]==a[y]?x<y:a[x]<a[y];}
inline int max(int x,int y){return x>y?x:y;}
inline int min(int x,int y){return x<y?x:y;}
void add(int p,int l,int r,int k,int v){
    z[p]=max(z[p],v);
    if (l==r) return;
    int mid=l+r>>;
    if (k<=mid) add(lp,l,mid,k,v);else add(rp,mid+,r,k,v);
}
int ask(int p,int l,int r,int L,int R){
    if (l==L&&r==R) return z[p];
    int mid=l+r>>;
    if (R<=mid) return ask(lp,l,mid,L,R);else
    if (L>mid) return ask(rp,mid+,r,L,R);else
    return max(ask(lp,l,mid,L,mid),ask(rp,mid+,r,mid+,R));
}
inline bool ju(int k){
    long long o=;
    int la=,e=;
    memset(z,,(n+)*);
    for (int i=;v[i].size();i++){
        for (int j=;j<v[i].size();){
            int w,m;
            for (w=j+;w<v[i].size();w++)
            if (v[i][w]-v[i][w-]>k) break;
            m=ask(,,n,max(,v[i][j]-k),min(n,v[i][w-]+k))+;
            for (;j!=w;j++)    o+=m,add(,,n,v[i][j],m);
        }
    }
    return o<=s;
}
int main(){
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&s);
        for (int i=;i<n;i++) scanf("%d",&a[i]),o[i]=i;
        sort(o,o+n,cmp);v[].push_back(o[]+);
        for (int i=,j=;i<n;i++) j+=a[o[i]]!=a[o[i-]],v[j].push_back(o[i]+);
        int l=,r=n;
        while (l<r){
            int mid=l+r+>>;
            if (ju(mid)) l=mid;else r=mid-;
        }
        printf("%d\n",ju(l)?l+:);
        for (int i=;i<n;i++) v[i].clear();
    }
}