【bzoj 4071】巴邻旁之桥（权值线段树）

传送门biu~
首先把S/T在河流同一侧的情况加到答案里，对于剩下的必须过桥的S/T分K=1和K=2两种情况：
当K=1时，只能建一座桥。显然是选择所有必须过桥的S和T值的中位数。
当K=2时，可以建两座桥。我们发现，对于一组S/T，假设S和T之间没有桥，那么如果走左边的桥pos距离为(S-pos)+(T-pos)+1；如果走右边的桥pos距离为(pos-S)+(pos-T)+1。所以真正决定走左桥还是右桥的是(S+T)/2到桥的距离哪个近，那么如果将所有的S/T按照(S+T)/2的大小排序，那么一定是以某个点为分界线，比它小的都走左边的桥，比它大的都走右边的桥。那么我们只需要枚举这个分界点，把这个点两边都分别看做一个K=1的情况就可以了，用【bzoj 1112】砖块Klo同样的方式用权值线段树动态维护中位数可以降低复杂度。
注意此题可能不存在S/T在河流两侧的情况，别问我是怎么知道的。

#include<bits/stdc++.h>
#define N 100005
#define inf 1000000000
using namespace std;
struct Node{
    Node *ch[2];
    Node();
    int siz;
    long long sum;
    inline void maintain(){
        sum=ch[0]->sum+ch[1]->sum;
        siz=ch[0]->siz+ch[1]->siz;
    }
}*null=new Node,*rootL=null,*rootR=null;
Node :: Node(){
    ch[0]=ch[1]=null;
    siz=sum=0;
}
void add(Node *&o,int l,int r,int x,int v){
    if(o==null)     o=new Node;
    if(l==r){
        o->sum+=x,o->siz+=v;
        return;
    }
    int mid=l+r>>1;
    if(abs(x)<=mid)     add(o->ch[0],l,mid,x,v);
    else    add(o->ch[1],mid+1,r,x,v);
    o->maintain();
}
int Kth(Node *o,int l,int r,int k){
    if(l==r)    return l;
    int mid=l+r>>1;
    if(o->ch[0]->siz>=k)    return Kth(o->ch[0],l,mid,k);
    else    return Kth(o->ch[1],mid+1,r,k-o->ch[0]->siz);
}
long long Query(Node *o,int l,int r,int p){
    if(o==null)     return 0;
    if(r<=p || l>p)     return abs(o->sum-1ll*p*o->siz);
    int mid=l+r>>1;
    return Query(o->ch[0],l,mid,p)+Query(o->ch[1],mid+1,r,p);
}
struct data{
    int x,y;
    data(){}
    data(int _,int __){x=_;y=__;}
}p[N];
inline bool cmp(data a,data b){return a.x+a.y<b.x+b.y;}
int n,k,a[2*N],tp;
long long ans;
int main(){
    scanf("%d%d",&k,&n);
    if(k==1){
        for(int i=1;i<=n;++i){
            char opt1[3],opt2[3];
            int x,y;
            scanf("%s%d%s%d",opt1,&x,opt2,&y);
            if(opt1[0]==opt2[0])    ans+=abs(x-y);
            else    ++ans,a[++tp]=x,a[++tp]=y;
        }
        if(!tp){
            printf("%lld\n",ans);
            return 0;
        }
        sort(a+1,a+tp+1);
        int med=a[tp/2];
        for(int i=1;i<=tp;++i)  ans+=abs(a[i]-med);
        printf("%lld\n",ans);
    }
    else{
        for(int i=1;i<=n;++i){
            char opt1[3],opt2[3];
            int x,y;
            scanf("%s%d%s%d",opt1,&x,opt2,&y);
            if(opt1[0]==opt2[0])    ans+=abs(x-y);
            else    ++ans,p[++tp]=data(x,y),add(rootR,0,inf,x,1),add(rootR,0,inf,y,1);
        }
        if(!tp){
            printf("%lld\n",ans);
            return 0;
        }
        sort(p+1,p+tp+1,cmp);
        int med=Kth(rootR,0,inf,tp);
        long long Min=Query(rootR,0,inf,med);
        for(int i=1;i<tp;++i){
            add(rootR,0,inf,-p[i].x,-1),add(rootR,0,inf,-p[i].y,-1);
            add(rootL,0,inf,p[i].x,1),add(rootL,0,inf,p[i].y,1);
            int medL=Kth(rootL,0,inf,i),medR=Kth(rootR,0,inf,tp-i);
            Min=min(Min,Query(rootL,0,inf,medL)+Query(rootR,0,inf,medR));
        }
        ans+=Min;
        printf("%lld\n",ans);
    }
    return 0;
}