bzoj1112: [POI2008]磚塊Klo（主席樹）

題目傳送門
。

解法：
比賽的時候不知道怎麼就一直在想主席樹（可能最近做主席樹有點多）
然後什麼線段樹啊樹狀陣列都沒想過。
然後就被師兄強行40M卡空間全爆了。。
大視野還是有良心的啊162M。。

就用主席樹求出中位數，然後沒了。

程式碼實現：

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
struct node {int c,lc,rc;ll s;}t[2000010];int cnt,rt[110000];
void build(int &u,int l,int r,int p,int s) {
    if(u==0)u=++cnt;t[u].c++;t[u].s+=(ll)s;
    if(l==r)return ;int mid=(l+r)/2;
    if(p<=mid)build(t[u].lc,l,mid,p,s);
    else build(t[u].rc,mid+1,r,p,s);
}
void Merge(int &u1,int u2) {
    if(u1==0) {u1=u2;return ;}if(u2==0)return ;
    t[u1].c+=t[u2].c;t[u1].s+=t[u2].s;
    Merge(t[u1].lc,t[u2].lc);Merge(t[u1].rc,t[u2].rc);
}
int find_mid(int &u1,int &u2,int l,int r,int k) {
    if(l==r)return l;int mid=(l+r)/2;
    int c=t[t[u1].lc].c-t[t[u2].lc].c;
    if(k<=c)return find_mid(t[u1].lc,t[u2].lc,l,mid,k);
    else return find_mid(t[u1].rc,t[u2].rc,mid+1,r,k-c);
}
int find_lc(int &u1,int &u2,int l,int r,int p) {
    if(t[u1].c-t[u2].c==0)return 0;
    if(l==r) return t[u1].c-t[u2].c;int mid=(l+r)/2;
    if(p<=mid)return find_lc(t[u1].lc,t[u2].lc,l,mid,p);
    else return find_lc(t[u1].rc,t[u2].rc,mid+1,r,p)+t[t[u1].lc].c-t[t[u2].lc].c;
}
ll find_ls(int &u1,int &u2,int l,int r,int p) {
    if(t[u1].s-t[u2].s==0) return 0;
    if(l==r)return t[u1].s-t[u2].s;int mid=(l+r)/2;
    if(p<=mid)return find_ls(t[u1].lc,t[u2].lc,l,mid,p);
    else return find_ls(t[u1].rc,t[u2].rc,mid+1,r,p)+t[t[u1].lc].s-t[t[u2].lc].s;
}
struct edge {int x,id;}a[110000];ll s[110000];
bool cmp(edge n1,edge n2){return n1.x<n2.x;}
bool cmp1(edge n1,edge n2){return n1.id<n2.id;}
int main() {
    //freopen("akc.in","r",stdin);freopen("akc.out","w",stdout);
    int n,k;scanf("%d%d",&n,&k);int mmax=0;
    for(int i=1;i<=n;i++) {scanf("%d",&a[i].x);a[i].id=i;}
    if(k==1){printf("0\n");return 0;}
    sort(a+1,a+1+n,cmp);int tot=0;a[0].x=-1;int tt=0;
    for(int i=1;i<=n;i++) if(a[i].x!=a[i-1].x)tt++;
    for(int i=1;i<=n;i++) {if(a[i].x!=a[i-1].x)tot++;build(rt[a[i].id],1,tt,tot,a[i].x);s[tot]=a[i].x;}
    for(int i=1;i<=n;i++) Merge(rt[i],rt[i-1]);
    sort(a+1,a+1+n,cmp1);ll T=0;for(int i=1;i<k;i++)T+=ll(a[i].x);ll Ans=ll(100000000000);a[0].x=0;
    for(int i=k;i<=n;i++) {
        T+=ll(a[i].x);T-=ll(a[i-k].x);
        int p=find_mid(rt[i],rt[i-k],1,tt,(k+1)/2);
        int lc=find_lc(rt[i],rt[i-k],1,tt,p),rc=k-lc;
        ll ls=find_ls(rt[i],rt[i-k],1,tt,p),rs=T-ls;
        Ans=min(Ans,ll(lc*s[p])-ls+rs-ll(rc*s[p]));
    }printf("%lld\n",Ans);
    return 0;
}