线段树 水题----今天先写点初学者的基础

hdu 1556
上来就可以直接讲题

省略头文件
using namespace std;
#define MAXN 100005
struct ST
{
    int l,r,sum;  //线段树的节点 l,r就是端点,sum用来记录被上过多少次色
}st[MAXN*3];
int nn;
void build(int L,int R,int n)    //建树的函数,一看就能明白,递归而已没什么难点。
{
    st[n].l=L;
    st[n].r=R;
    st[n].sum=0;
    if (L==R) return ;
    int mid=(L+R)/2;
    build(L,mid,2*n);
    build(mid+1,R,2*n+1);
};
void color(int L,int R,int n){  //注意:这里是把,st[n]的[l,r]上色,那么最后算结果的时候 
                                 //[l,r]往后的左右子树都要加上st[n].sum 
    if (st[n].l==L&&st[n].r==R) {st[n].sum++;return ;}
    int mid=(st[n].l+st[n].r)/2;
    if (R<=mid) color(L,R,2*n);
    else
    if (L>=mid+1) color(L,R,2*n+1);
    else                         // 如果L,R是被mid 分开的,那我们分别上色[L,mid]和[mid+1,R]就好了
    {
        color(L,mid,2*n);
        color(mid+1,R,2*n+1);
    }
}
void out(int n,int sum)          //输出函数,正如上面所说,单个节点的上色次数一定要加上父亲节点的上色次数
{
    if (st[n].l==st[n].r) {printf("%d",sum+st[n].sum);if (st[n].l!=nn) printf(" ");return;}
    out(2*n,sum+st[n].sum);
    out(2*n+1,sum+st[n].sum);
}
int main()
{
    //freopen("1.txt","r",stdin);
    int i,p,q;
    while (scanf("%d",&nn)&&nn)
    {
        build(1,nn,1);        //先建树
        for (i=1;i<=nn;++i) 
        {
            scanf("%d%d",&p,&q);
            color(p,q,1);   
        }
        out(1,0);
        printf("\n");
    }
    return 0;
}

你可能感兴趣的:(线段树)