bzoj1112 [POI2008]砖块Klo 线段树

Description

---

N柱砖,希望有连续K柱的高度是一样的. 你可以选择以下两个动作 1:从某柱砖的顶端拿一块砖出来,丢掉不要了. 2:从仓库中拿出一块砖,放到另一柱.仓库无限大. 现在希望用最小次数的动作完成任务.

第一行给出N,K. (1 ≤ k ≤ n ≤ 100000), 下面N行,每行代表这柱砖的高度.0 ≤ hi ≤ 1000000

Solution

---

刷水题，下午被一道卡常题卡傻了

答案显然为中位数。考虑枚举区间右端点，那么我们需要兹磁维护插入、删除、求中位数、求和
权值线段树或平衡树都可以

Code

---

#include <stdio.h>
#include <string.h>
#include <math.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)

typedef long long LL;
const LL INF=1e13;
const int N=200005;

struct treeNode {
    int l,r; LL sum,size;
} t[N*21];

int h[N],tot;

int read() {
    int x=0,v=1; char ch=getchar();
    for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
    for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
    return x*v;
}

void modify(int &now,int tl,int tr,int x,int v) {
    if (!now) now=++tot;
    t[now].size+=v;
    if (tl==tr) {
        t[now].sum=tl*1LL*t[now].size;
        return ;
    }
    int mid=(tl+tr)>>1;
    if (x<=mid) modify(t[now].l,tl,mid,x,v);
    else modify(t[now].r,mid+1,tr,x,v);
    t[now].sum=t[t[now].l].sum+t[t[now].r].sum;
}

LL query_sum(int now,int tl,int tr,int l,int r) {
    if (!now||r<l) return 0;
    if (tl==l&&tr==r) return t[now].sum;
    int mid=(tl+tr)>>1;
    if (r<=mid) return query_sum(t[now].l,tl,mid,l,r);
    if (l>mid) return query_sum(t[now].r,mid+1,tr,l,r);
    return query_sum(t[now].l,tl,mid,l,mid)+query_sum(t[now].r,mid+1,tr,mid+1,r);
}

LL query_size(int now,int tl,int tr,int l,int r) {
    if (!now||r<l) return 0;
    if (tl==l&&tr==r) return t[now].size;
    int mid=(tl+tr)>>1;
    if (r<=mid) return query_size(t[now].l,tl,mid,l,r);
    if (l>mid) return query_size(t[now].r,mid+1,tr,l,r);
    return query_size(t[now].l,tl,mid,l,mid)+query_size(t[now].r,mid+1,tr,mid+1,r);
}

int kth(int now,int tl,int tr,int k) {
    if (tl==tr) return tl;
    int mid=(tl+tr)>>1;
    if (t[t[now].l].size>=k) return kth(t[now].l,tl,mid,k);
    return kth(t[now].r,mid+1,tr,k-t[t[now].l].size);
}

int main(void) {
    freopen("data.in","r",stdin);
    freopen("myp.out","w",stdout);
    int n=read(),k=read(); LL ans=INF;
    int root=0;
    rep(i,1,n) h[i]=read();
    int lim=1000000;
    rep(i,1,k-1) modify(root,0,lim,h[i],1);
    rep(i,k,n) {
        modify(root,0,lim,h[i],1);
        LL pos=kth(root,0,lim,k/2+1);
        LL now=pos*query_size(root,0,lim,0,pos-1)-query_sum(root,0,lim,0,pos-1);
        now+=query_sum(root,0,lim,pos+1,lim)-pos*query_size(root,0,lim,pos+1,lim);
        ans=std:: min(ans,now);
        modify(root,0,lim,h[i-k+1],-1);
    }
    printf("%lld\n", ans);
    return 0;
}