【51NOD 1297】管理二叉树

Description

有一颗初始为空的二叉查找树，每次加入一个数，求每次加入后当前树中，两两点之间的距离和。
$n<=100000$

Solution

先把树给建出来，
暴力建树显然会被卡，

有结论：设当前的数为x，
比x小的数中最大的数是a，比x大的数中最大的是b，
a的右儿子和b的z左儿子，其中必定：有一个为空，另一个不为空；
那么，x的位置一定是在空的那个数下面。
证明：x一定在a的右儿子或以下，也一定在b的左儿子或以下。

建完树以后就直接上点分治即可。

复杂度：$O(n\log(n)^2)$（可以优化到一个log）

Code

这个程序可以加的优化很多。。。（所以跑的超级慢）

#include <iostream>
#include <cstdio>
#include <cstdlib>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define NX(q) ((q)&(-(q)))
using namespace std;
typedef long long LL;
const int N=100500;
int read(int &n)
{
    char ch=' ';int q=0,w=1;
    for(;(ch!='-')&&((ch<'0')||(ch>'9'));ch=getchar());
    if(ch=='-')w=-1,ch=getchar();
    for(;ch>='0' && ch<='9';ch=getchar())q=q*10+ch-48;n=q*w;return n;
}
int m,n;
int B[N][3];
int sm[N*3],sc[N];
LL ans[N];
bool z[N];
struct qqww
{
    int v,di,co;
}a[N];
int findl(int l,int r,int e,int r1)
{
    if(!sm[e])return 0;if(l==r)return l;
    int t=(l+r)>>1;
    if(r1<=t)return findl(l,t,e*2,r1);
    int ans=findl(t+1,r,e*2+1,r1);
    return ans?ans:findl(l,t,e*2,r1);
}
int findr(int l,int r,int e,int l1)
{
    if(!sm[e])return 0;if(l==r)return l;
    int t=(l+r)>>1;
    if(l1>t)return findr(t+1,r,e*2+1,l1);
    int ans=findr(l,t,e*2,l1);
    return ans?ans:findr(t+1,r,e*2+1,l1);
}
void change(int l,int r,int e,int l1)
{
    if(l==r){sm[e]++;return;}
    int t=(l+r)>>1;
    if(l1<=t)change(l,t,e*2,l1);
        else if(t<l1)change(t+1,r,e*2+1,l1);
    sm[e]=sm[e*2]+sm[e*2+1];
}
int Hv,Hvs;
LL f[2][N],f1[2][N];
int dfsh(int q,int fa,int w)
{
    a[q].co=1;
    int mx=0;
    fo(i,0,2)if(!z[B[q][i]]&&B[q][i]!=fa)
    {
        int t=dfsh(B[q][i],q,w);
        a[q].co+=t;mx=max(mx,t);
    }
    mx=max(mx,w-a[q].co);
    if(mx<Hvs)Hvs=mx,Hv=q;
    return a[q].co;
}
void add(int q,LL di,int I,int v)
{
    for(int i=q;i<=n;i+=NX(i))f[I][i]+=di*v,f1[I][i]+=v;
}
LL Rt,Rt1;
void Gsum(int q,int I,int v)
{
    for(int i=q;i>0;i-=NX(i))Rt+=f[I][i]*v,Rt1+=f1[I][i]*v;
}
void dfsf(int q,int fa,LL di,int I,int v)
{
    fo(i,0,2)if(!z[B[q][i]]&&B[q][i]!=fa)dfsf(B[q][i],q,di+1,I,v);
    add(a[q].v,di,I,v);
}
void dfs(int q,int fa,LL di)
{
    Rt=Rt1=0;Gsum(a[q].v,0,1);
    Gsum(a[q].v,1,-1);
    ans[q]+=Rt+Rt1*di;
    fo(i,0,2)if(!z[B[q][i]]&&B[q][i]!=fa)dfs(B[q][i],q,di+1);
}
void divide(int q,int alln)
{
    if(alln==1)return;
    Hv=Hvs=n+1;
    dfsh(q,0,alln);
    q=Hv;z[q]=1;
    dfsf(q,0,0,0,1);
    Rt=Rt1=0;Gsum(a[q].v,0,1);
    ans[q]+=Rt;
    fo(i,0,2)if(!z[B[q][i]])
    {
        dfsf(B[q][i],q,1,1,1);
        dfs(B[q][i],q,1);
        dfsf(B[q][i],q,1,1,-1);
    }
    dfsf(q,0,0,0,-1);
    fo(i,0,2)if(!z[B[q][i]])divide(B[q][i],(a[q].co>a[B[q][i]].co)?a[B[q][i]].co:alln-a[q].co);
}
int main()
{
    int q,w,t;
    read(n);
    fo(i,1,n)
    {
        sc[i]=read(t);
        q=findl(1,n,1,t);
        w=findr(1,n,1,t);
        if(a[w].v>a[q].v)q=w;
        B[t][0]=q;
        a[t].v=i;change(1,n,1,t);
    }
    fo(i,1,n)if(B[i][0])B[B[i][0]][(B[i][0]<i)+1]=i;
    z[0]=1;
    divide(1,n);
    fo(i,1,n)ans[sc[i]]+=ans[sc[i-1]],printf("%lld\n",ans[sc[i]]);
    return 0;
}