线段树poj3667

线段树

Query，查找符合题意得最左端的点，当tree[t<<1].sum<x时，若tree[t<<1].rsum+tree[t<<1|1].lsum>=x，就返回左儿子rsum的起始点，否则递归走向右儿子

Updata  用来重置树，当tree[t].l==l&&tree[t].r==r时重置，后返回，用push_up将所有父节点重置。用push_down将子节点重置。

代码有参考侵删

#include<stdio.h>
#include<string.h>
#define Max 4*100000
#define Mid(x,y)(x+y)>>1//(x+y)/2
int A[Max];
struct hello
{
    int l;
    int r;
    int rsum;
    int lsum;
    int sum;
    int loop;
} tree[Max];
int Find_Max(int a,int b,int c)
{
    int sum=a;
    if(b>sum)
        sum=b;
    if(c>sum)
        sum=c;
    return sum;


}
void Push_down(int t)
{
    if(tree[t].loop!=-1)
    {
        tree[t<<1].loop=tree[t<<1|1].loop=tree[t].loop;
        if(tree[t].loop)
        {
            tree[t<<1].lsum=tree[t<<1].rsum=tree[t<<1].sum=0;
            tree[t<<1|1].lsum=tree[t<<1|1].rsum=tree[t<<1|1].sum=0;
        }
        else
        {
            tree[t<<1].lsum=tree[t<<1].rsum=tree[t<<1].sum=tree[t<<1].r-tree[t<<1].l+1;
            tree[t<<1|1].lsum=tree[t<<1|1].rsum=tree[t<<1|1].sum=tree[t<<1|1].r-tree[t<<1|1].l+1;
        }
        tree[t].loop=-1;
    }
}
void Push_up(int l,int r,int t)
{
    tree[t].lsum=tree[t<<1].lsum;
    tree[t].rsum=tree[t<<1|1].rsum;
    int x=(r+l)/2;
    if(tree[t].lsum==x-l+1)
        tree[t].lsum+=tree[t<<1|1].lsum;
    if(tree[t].rsum==r-x)
        tree[t].rsum+=tree[t<<1].rsum;
    tree[t].sum=Find_Max(tree[t<<1].sum,tree[t<<1|1].sum,tree[t<<1].rsum+tree[t<<1|1].lsum);
}
void Build_tree(int l,int r,int t)// l,r 表示区间，t表示 区间节点
{
    tree[t].l=l;
    tree[t].r=r;
    tree[t].sum=tree[t].lsum=tree[t].rsum=r-l+1;
    tree[t].loop=-1;
    if(l==r)
        return;
    int x=Mid(l,r);
    Build_tree(l,x,2*t);
    Build_tree(x+1,r,2*t+1);
}
void Updata_tree(int l,int r,int t,int cnt)
{
    if(tree[t].l==l&&tree[t].r==r)
    {
        tree[t].loop=cnt;
        if(cnt)
        {
            tree[t].lsum=tree[t].rsum=tree[t].sum=0;
        }
        else
        {
            tree[t].lsum=tree[t].rsum=tree[t].sum=r-l+1;
        }
        return;
    }
    Push_down(t);
    int x=Mid(tree[t].l,tree[t].r);
    if(x>=r)
        Updata_tree(l,r,2*t,cnt);
    else if(x+1<=l)
        Updata_tree(l,r,2*t+1,cnt);
    else
    {
        Updata_tree(l,x,2*t,cnt);
        Updata_tree(x+1,r,2*t+1,cnt);
    }
    Push_up(tree[t].l,tree[t].r,t);
}
int Query_tree(int l,int r,int t,int cnt)
{
    if(l==r)
        return l;
    int x=Mid(l,r);
    Push_down(t);
    if(tree[t<<1].sum>=cnt)
        return Query_tree(l,x,2*t,cnt);
    else if(tree[t<<1].rsum+tree[t<<1|1].lsum>=cnt)
        return x-tree[t<<1].rsum+1;
    else
        return Query_tree(x+1,r,2*t+1,cnt);
}
int main()
{
    int i,j,n,m,x,y;
    while(~scanf("%d%d",&n,&m))
    {
        Build_tree(1,n,1);
        while(m--)
        {
            scanf("%d",&i);
            if(i==1)
            {
                scanf("%d",&x);
                if(tree[1].sum<x)
                {
                    printf("0\n");
                    continue;
                }
                y=Query_tree(1,n,1,x);
                printf("%d\n",y);
                Updata_tree(y,y+x-1,1,1);
            }
            else
            {
                scanf("%d%d",&x,&y);
                Updata_tree(x,x+y-1,1,0);
            }
        }
    }
    return 0;
}
/*
10 6
1 3
1 3
1 3
1 3
2 5 5
1 6
*/