SGU 263 Towers(树状数组)

题目链接：http://acm.sgu.ru/problem.php?contest=0&problem=263

题意：一排位置，开始全为0.若一段区间[L,R]之间均大于0且L-1和R+1均为0，我们说[L,R]为一段。两种改变，四种询问。改变是：（1）将第x个位置增加y；（2）将第t段的第x个位置增加y。四种询问是：（1）段的个数；（2）第t段中的所有数之和；（3）第t段的区间长度；（4）第t段上第x个位置的数是多少。

思路：一开始用splay+并查集WA了半天。最后看别人的BIT+并查集，好犀利。。。

int root[N],num[N],c[N],sum[N],p[N];
int n;

int getRoot(int x)
{
    if(root[x]!=x) root[x]=getRoot(root[x]);
    return root[x];
}

int search(int c[],int x,int y)
{
    int ans=0;
    while(y) ans+=c[y],y-=y&-y;
    x--;
    while(x) ans-=c[x],x-=x&-x;
    return ans;
}

void insert(int c[],int x,int det)
{
    while(x<N) c[x]+=det,x+=x&-x;
}


void Union(int x,int y)
{
    num[x]+=num[y];
    root[y]=x;
}

int getStart(int k)
{
    int low=1,high=N-1,mid,temp;
    while(low<=high)
    {
        mid=(low+high)>>1;
        if(search(c,1,mid)<k) low=mid+1;
        else high=mid-1;
    }
    if(high>=1&&search(c,1,high)==k) return high;
    return low;
}

void putCube(int x,int y)
{
    insert(sum,y,x);
    if(p[y]) return;
    p[y]=1;
    x=getRoot(y-1);
    if(p[x]) Union(x,y);
    else insert(c,y,1);
    if(p[y+1])
    {
        insert(c,y+1,-1);
        x=getRoot(y);
        Union(x,y+1);
    }
}

void put()
{
    int x,c;
    scanf("%d%d",&x,&c);
    putCube(c,x);
}

void tput()
{
    int t,x,c,p;
    scanf("%d%d%d",&t,&x,&c);
    p=getStart(t);
    putCube(c,p+x-1);
}

void towers()
{
    printf("%d towers\n",search(c,1,N-1));
}

void cubes()
{
    int t,p;
    scanf("%d",&t);
    p=getStart(t);
    printf("%d cubes in %dth tower\n", search(sum,p,p+num[p]-1),t);
}

void length()
{
    int t,p;
    scanf("%d",&t);
    p=getStart(t);
    printf("length of %dth tower is %d\n",t,num[p]);
}

void tcubes()
{
    int t,x,p;
    scanf("%d%d",&t,&x);
    p=getStart(t)+x-1;
    printf("%d cubes in %dth column of %dth tower\n",search(sum,p,p),x,t);
}

int main()
{
    int i;
    for(i=0;i<N;i++) num[i]=1,root[i]=i;
    scanf("%d",&n);
    char cmd[15];
    while(n--)
    {
        scanf("%s",cmd);
        if(cmd[0]=='p') put();
        else if(cmd[0]=='t'&&cmd[1]=='p') tput();
        else if(cmd[0]=='t'&&cmd[1]=='o') towers();
        else if(cmd[0]=='c') cubes();
        else if(cmd[0]=='l') length();
        else if(cmd[0]=='t'&&cmd[1]=='c') tcubes();
    }
}