[BZOJ 4071] 巴邻旁之桥

Link:

BZOJ 4071传送门

Solution:

首先算出能提前算的贡献

$K=1$：肯定选中间的点，小学数学

$K=2$：对于每对$(x,y)$一定选离$(x+y)/2$近的桥

也就是说将$(x,y)$按$(x+y)/2$的值排序后一定恰有一个分割点使得两边选择不同的桥！

考虑如何如何快速枚举所有分割点时的答案：

需要支持插入、删除、求中位数及两边的和，明显选择$Splay$来维护（求和更容易些）

这样先将所有数对加进一个$Splay$，再不断将其中的数移向另一棵$Splay$即可

注意：可能以前的$Splay$板子容易出不少锅啊……

首先$Update$时最好对$z$进行判断，然后$Splay$最后也要$Pushup$否则会在树高低时不更新$size$

Code:

#include <bits/stdc++.h>
 
using namespace std;
#define X first
#define Y second
typedef double db;
typedef long long ll;
typedef pair<int,int> P;
const int MAXN=2e5+;//注意范围
const ll INF=<<;
 
struct Splay
{
    ll sum[MAXN];
    int rt,tot,f[MAXN],sz[MAXN],cnt[MAXN],ch[MAXN][],val[MAXN];
    void update(int x)
    {
        sz[x]=sz[ch[x][]]+sz[ch[x][]]+cnt[x];
        sum[x]=sum[ch[x][]]+sum[ch[x][]]+1ll*val[x]*cnt[x];
    }
    void rotate(int x)
    {
        int y=f[x],z=f[y],k=(ch[y][]==x);
        //最好对z特判一下防止对ch[0]操作
        if(z) ch[z][ch[z][]==y]=x;f[x]=z;
        ch[y][k]=ch[x][k^];f[ch[x][k^]]=y;
        ch[x][k^]=y;f[y]=x;
        update(y);update(x);
    }
    void splay(int x,int up)
    {
        while(f[x]!=up)
        {
            int y=f[x],z=f[y];
            if(z!=up) (ch[y][]==x)^(ch[z][]==y)?rotate(x):rotate(y);
            rotate(x);
        }
        if(!up) rt=x;
        //!!!!!!!!
        update(x);//可能f[x]=up，但也要update
    }
    void insert(int x)
    {
        int k=rt,anc=;
        while(k&&val[k]!=x)
            anc=k,k=ch[k][x>val[k]];
        if(k) cnt[k]++;
        else
        {
            k=++tot;
            if(anc) ch[anc][x>val[anc]]=k;
            val[k]=x;cnt[k]=;f[k]=anc;sz[k]=;
        }
        splay(k,);
    }
    void find(int x)
    {
        int k=rt;
        while(x!=val[k]&&ch[k][x>val[k]])
            k=ch[k][x>val[k]];
        splay(k,);
    }
    int kth(int x)
    {
        int k=rt;
        //对当前为空的情况特判
        if(sz[k]<x||!x) return ;
        while(true)
        {
            if(x>sz[ch[k][]]+cnt[k])
                x-=sz[ch[k][]]+cnt[k],k=ch[k][];
            else if(x<=sz[ch[k][]]) k=ch[k][];
            else return k;
        }
    }
    //不用加边界的删除法
    void del(int x)
    {
        find(x);
        if(val[rt]!=x) return;
        if(cnt[rt]>) cnt[rt]--;
        else if(!ch[rt][]||!ch[rt][])
        {
            int k=ch[rt][]+ch[rt][];
            f[k]=;rt=k;
        }
        else
        {
            int k=ch[rt][];
            while(ch[k][]) k=ch[k][];
            splay(k,rt);
            ch[k][]=ch[rt][];
            f[ch[k][]]=k;
            rt=k;f[k]=;
        }
        //由于后面没有splay了要update
        update(rt);
    }
    ll query()
    {
        int k=kth(sz[rt]/);
        if(!k) return ;splay(k,);
        ll ret1=1ll*val[k]*sz[ch[k][]]-sum[ch[k][]];
        ll ret2=sum[ch[k][]]-1ll*val[k]*sz[ch[k][]];
        return ret1+ret2;
    }
}s1,s2;
 
char a[],b[];
ll res=1ll<<,pre;
int n,x,y,k,tot;P dat[MAXN];
 
bool cmp(P a,P b){return a.X+a.Y<b.X+b.Y;};
 
int main()
{
    scanf("%d%d",&k,&n);
    for(int i=;i<=n;i++)
    {
        scanf("%s%d%s%d",a,&x,b,&y);
        if(a[]==b[]) pre+=abs(x-y);
        else dat[++tot]=P(x,y),pre++;
    }
    //注意特判！！！
    if(!tot) return printf("%lld",pre),;
    sort(dat+,dat+tot+,cmp);
 
    for(int i=;i<=tot;i++)
        s1.insert(dat[i].X),s1.insert(dat[i].Y);
    if(k==) return printf("%lld",s1.query()+pre),;
 
    for(int i=;i<=tot;i++)
    {
        s1.del(dat[i].X);s1.del(dat[i].Y);
        s2.insert(dat[i].X);s2.insert(dat[i].Y);
        res=min(res,s1.query()+s2.query());
    }
    printf("%lld",res+pre);
    return ;
}