FZU 2105 线段树经典题目

#include 
#include 
#include 
#include 
#define N 1000010
#define ls p<<1
#define rs p<<1|1
using namespace std;

struct node
{
    int l,r;
    int cn;
} aa[N<<2];

void bg(int l,int r,int p)
{
    aa[p].l = l;
    aa[p].r = r;
    aa[p].cn = -1;
    if(l==r)
    {
        scanf("%d",&aa[p].cn);
        return;
    }
    int mid = (l+r)>>1;
    bg(l,mid,ls);
    bg(mid+1,r,rs);
    if(aa[ls].cn!=-1&&aa[ls].cn==aa[rs].cn)
    {
        aa[p].cn = aa[ls].cn;
    }
}

int opp(int op,int opn,int cn)
{
    if(op==1)   return opn&cn;
    if(op==2)   return opn|cn;
    if(op==3)   return opn^cn;

}

void update(int l,int r,int p,int opn,int op)
{
    if(aa[p].l==l&&aa[p].r==r&&aa[p].cn>=0)
    {
        aa[p].cn = opp(op,opn,aa[p].cn);
        return;
    }
    if(aa[p].cn>=0)
    {
        aa[ls].cn = aa[rs].cn = aa[p].cn;
        aa[p].cn = -1;
    }
    int mid = (aa[p].l+aa[p].r)>>1;
    if(r<=mid)  update(l,r,ls,opn,op);
    else if(l>mid)  update(l,r,rs,opn,op);
    else
    {
        update(l,mid,ls,opn,op);
        update(mid+1,r,rs,opn,op);
    }
    if(aa[ls].cn!=-1&&aa[ls].cn==aa[rs].cn)  //向上更新
    {
        aa[p].cn = aa[ls].cn;
    }
}

__int64 query(int l,int r,int p)
{
    if(aa[p].l==l&&aa[p].r==r&&aa[p].cn>=0)
    {
        return aa[p].cn*(aa[p].r-aa[p].l+1);
    }
    if(aa[p].cn>=0)  //经过该区间时,向下更新
    {
        aa[ls].cn = aa[rs].cn = aa[p].cn;
        aa[p].cn = -1;
    }
    int mid = (aa[p].l+aa[p].r)>>1;
    if(r<=mid)  return query(l,r,ls);
    else if(l>mid)  return query(l,r,rs);
    else return query(l,mid,ls)+query(mid+1,r,rs);
    if(aa[ls].cn!=-1&&aa[ls].cn==aa[rs].cn)  //向上更新
    {
        aa[p].cn = aa[ls].cn;
    }
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n,m;
        scanf("%d%d",&n,&m);
        char op[5];
        bg(1,n,1);
        while(m--)
        {
            scanf("%s",op);
            int opn,a,b;
            if(op[0]=='S')
            {
                scanf("%d%d",&a,&b);
                printf("%I64d\n",query(a+1,b+1,1));
            }
            else
            {
                scanf("%d%d%d",&opn,&a,&b);
                if(op[0]=='A')
                    update(a+1,b+1,1,opn,1);
                else if(op[0]=='O')
                    update(a+1,b+1,1,opn,2);
                else if(op[0]=='X')
                    update(a+1,b+1,1,opn,3);
            }
        }
    }
    return 0;
}

你可能感兴趣的:(数据结构)