poj 2352 Stars 夜

http://poj.org/problem?id=2352

题目大意：

给出N个点 点的顺序是按y非递减给出的

level是这个点左下方点的个数

求各个level的数量

思路：

由于给的点有顺序 所以就简单了

一般用树状数组或线段树就可以

我用了平衡树 是为了练习啦 写的相当烂

用平衡树 不断地插入更新就可以了

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<queue>
#include<algorithm>

using namespace std;

const int N=15100;
struct node
{
    int x;//x坐标
    int l,r,f;//左右孩子 和父节点
    int sum;
    int h;
}mem[N];
int treehead;//树的头结点
int ans[N];//答案
void Leftturn(int i)//左旋转
{
    int f=mem[i].f;
    int r=mem[i].r;
    if(f==-1)
    {
        mem[r].f=-1;treehead=r;
    }
    else
    {
        mem[r].f=f;
        if(mem[f].l==i)
        mem[f].l=r;
        else
        mem[f].r=r;
    }
    mem[i].r=mem[r].l;
    mem[mem[r].l].f=i;
    mem[r].l=i;
    mem[i].f=r;
    mem[i].h=max(mem[mem[i].l].h,mem[mem[i].r].h)+1;
    mem[i].sum=mem[mem[i].l].sum+mem[mem[i].r].sum+1;
    mem[r].h=max(mem[mem[r].l].h,mem[mem[r].r].h)+1;
    mem[r].sum=mem[mem[r].l].sum+mem[mem[r].r].sum+1;
}
void Rightturn(int i)//右旋转
{
    int f=mem[i].f;
    int l=mem[i].l;
    if(f==-1)
    {
        mem[l].f=-1;treehead=l;
    }
    else
    {
        mem[l].f=f;
        if(mem[f].l==i)
        mem[f].l=l;
        else
        mem[f].r=l;
    }
    mem[i].l=mem[l].r;
    mem[mem[l].r].f=i;
    mem[l].r=i;
    mem[i].f=l;
    mem[i].h=max(mem[mem[i].l].h,mem[mem[i].r].h)+1;
    mem[i].sum=mem[mem[i].l].sum+mem[mem[i].r].sum+1;
    mem[l].h=max(mem[mem[l].l].h,mem[mem[l].r].h)+1;
    mem[l].sum=mem[mem[l].l].sum+mem[mem[l].r].sum+1;
}

void update(int i)//对不平衡点的更新
{
    if(mem[mem[i].l].h<mem[mem[i].r].h)
    {
        int x=mem[i].r;
        if(mem[mem[x].l].h<mem[mem[x].r].h)
        {
            Leftturn(i);
        }
        else
        {
            Rightturn(mem[i].r);
            Leftturn(i);
        }
    }
    else
    {
        int x=mem[i].l;
        if(mem[mem[x].l].h>mem[mem[x].r].h)
        {
            Rightturn(i);
        }
        else
        {
            Leftturn(mem[i].l);
            Rightturn(i);
        }
    }
}
void insert(int i,int k,int s)//插入新的点
{
    if(mem[i].x<=mem[k].x)
    {
        if(mem[i].r!=0)
        {
            insert(mem[i].r,k,s+mem[mem[i].l].sum+1);//继续向右插入 注意记录有多少比他小的点
        }
        else
        {
            ++ans[s+mem[mem[i].l].sum+1];
            mem[i].r=k;
            mem[k].f=i;
        }
    }
    else
    {
        if(mem[i].l!=0)
        {
            insert(mem[i].l,k,s);
        }
        else
        {
            ++ans[s];
            mem[i].l=k;
            mem[k].f=i;
        }
    }
    mem[i].h=max(mem[mem[i].l].h,mem[mem[i].r].h)+1;//对路径上的点要更新
    mem[i].sum=mem[mem[i].l].sum+mem[mem[i].r].sum+1;
    if(abs(mem[mem[i].l].h-mem[mem[i].r].h)>1)
    {
        update(i);
    }
}
int main()
{
    int n;
    scanf("%d",&n);
        memset(ans,0,sizeof(ans));
        mem[0].l=mem[0].r=-1;//把0这个点设成假的所以叶子节点
        mem[0].sum=0;
        mem[0].h=-1;
        int y;
        scanf("%d %d",&mem[1].x,&y);
        mem[1].h=0;
        mem[1].l=mem[1].r=0;
        mem[1].sum=1;mem[1].f=-1;
        treehead=1;//头结点
        ++ans[0];
        for(int i=2;i<=n;++i)
        {
            scanf("%d %d",&mem[i].x,&y);
            mem[i].l=mem[i].r=0;
            mem[i].sum=1;mem[i].h=0;
            insert(treehead,i,0);
        }
        for(int i=0;i<n;++i)
        {
            printf("%d\n",ans[i]);
        }
    return 0;
}