51 nod 1297 管理二叉树

原题链接：https://www.51nod.com/onlineJudge/questionCode.html#!problemId=1297

先是暴力加优化T了最后两个点……

我还是来想想正解吧。

先写棵线段树把二叉搜索树建出来，然后在树上做下点分治就好了。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define lp (p<<1)
#define rp ((p<<1)|1)
using namespace std;

int read_p,read_ca;
inline int read(){
    read_p=;read_ca=getchar();
    while(read_ca<''||read_ca>'') read_ca=getchar();
    while(read_ca>=''&&read_ca<='') read_p=read_p*+read_ca-,read_ca=getchar();
    return read_p;
}
struct na{int y,z,ne,k;}b[],B[];
int n,m,a[],x,y,l[],num=,de[],L[],Num=,s[],Si,ro,ma;
bool bo[];
long long mmh[][],S[][];
long long MMH=;
inline void in(int x,int y){b[++num].y=y;b[num].ne=l[x];l[x]=num;}
inline void In(int x,int y,int z,int k){B[++Num].y=y;B[Num].z=z;B[Num].k=k;B[Num].ne=L[x];L[x]=Num;}
struct tree{int ma,mi;}t[];
void add(int p,int l,int r,int v){
    if (a[t[p].ma]<a[v]||t[p].ma==) t[p].ma=v;
    if (a[t[p].mi]>a[v]||t[p].mi==) t[p].mi=v;
    if (l==r) return;
    int mid=l+r>>;
    if (a[v]<=mid) add(lp,l,mid,v);else add(rp,mid+,r,v);
}
int quea(int p,int l,int r,int R){
    if (R>=r) return t[p].ma;
    int mid=l+r>>;
    if (R<=mid) return quea(lp,l,mid,R);else if (mid=quea(rp,mid+,r,R)) return mid;else return t[lp].ma;
}
int quei(int p,int l,int r,int R){
    if (R<=l) return t[p].mi;
    int mid=l+r>>;
    if (R>mid) return quei(rp,mid+,r,R);else if (mid=quei(lp,l,mid,R)) return mid;else return t[rp].mi;
}
void gs(int x,int f){
    s[x]=;
    int u=;
    for (register int i=l[x];i;i=b[i].ne)
    if (b[i].y!=f&&!bo[b[i].y]){
        gs(b[i].y,x);
        s[x]+=s[b[i].y];
        if (u<s[b[i].y]) u=s[b[i].y];
    }
    if (Si-s[x]>u) u=Si-s[x];
    if (u<ma) ma=u,ro=x;
}
void dfs(int x,int f,int r,int z,int k){
    s[x]=;
    In(x,r,z,k);
    for (register int i=l[x];i;i=b[i].ne)
    if (!bo[b[i].y]&&b[i].y!=f) dfs(b[i].y,x,r,z+,k),s[x]+=s[b[i].y];
}
void work(int x,int S){
    Si=S;ma=1e9;gs(x,);bo[x=ro]=;
    int u=;
    for (register int i=l[x];i;i=b[i].ne) if (!bo[b[i].y]) dfs(b[i].y,x,x,,++u);In(x,x,,++u);
    for (register int i=l[x];i;i=b[i].ne) if (!bo[b[i].y]) work(b[i].y,s[b[i].y]);
}
int main(){
    register int i,j;
    n=read();a[n+]=n+;add(,,n+,);add(,,n+,n+);
    a[]=read();add(,,n+,);de[]=;
    for (i=;i<=n;i++){
        a[i]=read();
        x=quea(,,n+,a[i]-);y=quei(,,n+,a[i]+);
        if (de[x]>de[y]){
            in(x,i);
            in(i,x);
            de[i]=de[x]+;
        }else{
            in(y,i);
            in(i,y);
            de[i]=de[y]+;
        }
        add(,,n+,i);
    }
    work(,n);
    for (i=;i<=n;i++){
        for (j=L[i];j;j=B[j].ne) MMH+=1LL*B[j].z*(S[B[j].y][]-S[B[j].y][B[j].k])+mmh[B[j].y][]-mmh[B[j].y][B[j].k],
        mmh[B[j].y][]+=B[j].z,mmh[B[j].y][B[j].k]+=B[j].z,S[B[j].y][]++,S[B[j].y][B[j].k]++;
        printf("%lld\n",MMH);
    }
}