区间合并 成段异或更新 求最长01段

题目链接     hdu3397 Sequence operation

题目大意:有n个数,可能为0或为1。接下来是m个操作,操作有5种类型。(1)“0 a b”,表示将区间[a,b]范围内的数全部置0;(2)“1 a b”,表示将区间[a,b]内的数全部置1;(3)"2 a b",表示将区间[a,b]内的数0变成1,1变成0;(4)"3 a b",表示查询[a,b]范围内1的数;(5)"4 a b",表示查询[a,b]范围内最长的连续的1;

解题思路:开一个结构体,记录一个区间内0的个数sum0,1的个数sum1,lm0:从左端点开始最长的连续0,lm1:从左端点开始最长的连续1,rm0:以右端点结束最长的0,rm1:以右端点结束最长的1,sm0:整段区间最长的连续0,sm1:整段区间最长的连续1。

这里也要注意先覆盖再异或和先异或再覆盖的区别。先异或再覆盖会把异或标记给覆盖掉,先覆盖再异或则不能把之前的更新覆盖掉。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

#define lz 2*u,l,mid
#define rz 2*u+1,mid+1,r
const int maxn=100005;
int a[maxn];
int flag[4*maxn];

struct node
{
    int lm0, rm0, sm0;
    int lm1, rm1, sm1;
    int sum0, sum1;
} tree[4*maxn];

void Xor(int u)
{
    swap(tree[u].sum1,tree[u].sum0);
    swap(tree[u].lm1,tree[u].lm0);
    swap(tree[u].rm1,tree[u].rm0);
    swap(tree[u].sm1,tree[u].sm0);

    if(flag[u]==0) flag[u]=3;
    else if(flag[u]==1) flag[u]=2;
    else if(flag[u]==2) flag[u]=1;
    else if(flag[u]==3) flag[u]=0;
}

void push_up(int u, int l, int r)
{
    tree[u].sum1=tree[2*u].sum1+tree[2*u+1].sum1;
    tree[u].sum0=tree[2*u].sum0+tree[2*u+1].sum0;
    tree[u].lm0=tree[2*u].lm0;
    tree[u].lm1=tree[2*u].lm1;
    tree[u].rm0=tree[2*u+1].rm0;
    tree[u].rm1=tree[2*u+1].rm1;
    tree[u].sm0=max(tree[2*u].sm0,tree[2*u+1].sm0);
    tree[u].sm1=max(tree[2*u].sm1,tree[2*u+1].sm1);
    int mid=(l+r)>>1;

    if(tree[2*u].lm1==mid-l+1) tree[u].lm1+=tree[2*u+1].lm1;
    if(tree[2*u+1].rm1==r-mid) tree[u].rm1+=tree[2*u].rm1;
    int t=tree[2*u].rm1+tree[2*u+1].lm1;
    if(t>tree[u].sm1) tree[u].sm1=t;

    if(tree[2*u].lm0==mid-l+1) tree[u].lm0+=tree[2*u+1].lm0;
    if(tree[2*u+1].rm0==r-mid) tree[u].rm0+=tree[2*u].rm0;
    int h=tree[2*u].rm0+tree[2*u+1].lm0;
    if(h>tree[u].sm0) tree[u].sm0=h;
}

void push_down(int u, int l, int r)
{
    if(flag[u])
    {
        int mid=(l+r)>>1;
        if(flag[u]==1)
        {
            tree[2*u].sum1=tree[2*u].lm1=tree[2*u].rm1=tree[2*u].sm1=0;
            tree[2*u].sum0=tree[2*u].lm0=tree[2*u].rm0=tree[2*u].sm0=mid-l+1;

            tree[2*u+1].sum1=tree[2*u+1].lm1=tree[2*u+1].rm1=tree[2*u+1].sm1=0;
            tree[2*u+1].sum0=tree[2*u+1].lm0=tree[2*u+1].rm0=tree[2*u+1].sm0=r-mid;
            flag[2*u]=flag[2*u+1]=1;
        }
        else if(flag[u]==2)
        {
            tree[2*u].sum1=tree[2*u].lm1=tree[2*u].rm1=tree[2*u].sm1=mid-l+1;
            tree[2*u].sum0=tree[2*u].lm0=tree[2*u].rm0=tree[2*u].sm0=0;

            tree[2*u+1].sum1=tree[2*u+1].lm1=tree[2*u+1].rm1=tree[2*u+1].sm1=r-mid;
            tree[2*u+1].sum0=tree[2*u+1].lm0=tree[2*u+1].rm0=tree[2*u+1].sm0=0;
            flag[2*u]=flag[2*u+1]=2;
        }
        else
        {
            Xor(2*u);
            Xor(2*u+1);
        }
        flag[u]=0;
    }
}

void build(int u, int l, int r)
{
    flag[u]=0;
    if(l==r)
    {
        tree[u].sum1=tree[u].lm1=tree[u].rm1=tree[u].sm1=a[l]?1:0;
        tree[u].sum0=tree[u].lm0=tree[u].rm0=tree[u].sm0=a[l]?0:1;
        return ;
    }
    int mid=(l+r)>>1;
    build(lz);
    build(rz);
    push_up(u,l,r);
}

void Update(int u, int l, int r, int tl, int tr, int c)
{
    if(tl<=l&&r<=tr)
    {
        if(c==1)
        {
            tree[u].sum1=tree[u].lm1=tree[u].rm1=tree[u].sm1=0;
            tree[u].sum0=tree[u].lm0=tree[u].rm0=tree[u].sm0=r-l+1;
            flag[u]=1;
        }
        else if(c==2)
        {
            tree[u].sum1=tree[u].lm1=tree[u].rm1=tree[u].sm1=r-l+1;
            tree[u].sum0=tree[u].lm0=tree[u].rm0=tree[u].sm0=0;
            flag[u]=2;
        }
        else    /// 这里要特别注意,对于异或操作不能把此前的标记给覆盖掉,必须进行相应的更新才能保证向下延迟标记不出错,这里错了好久
        {
            Xor(u);
        }
        return ;
    }
    push_down(u,l,r);
    int mid=(l+r)>>1;
    if(tr<=mid) Update(lz,tl,tr,c);
    else if(tl>mid) Update(rz,tl,tr,c);
    else
    {
        Update(lz,tl,mid,c);
        Update(rz,mid+1,tr,c);
    }
    push_up(u,l,r);
}

int Query1(int u,int l, int r, int tl, int tr)
{
    if(tl<=l&&r<=tr)
    {
        return tree[u].sum1;
    }
    push_down(u,l,r);
    int mid=(l+r)>>1;
    if(tr<=mid) return Query1(lz,tl,tr);
    else if(tl>mid) return Query1(rz,tl,tr);
    else
    {
        int t1=Query1(lz,tl,mid);
        int t2=Query1(rz,mid+1,tr);
        return t1+t2;
    }
}

int Query2(int u, int l, int r, int tl, int tr)
{
    if(tl<=l&&r<=tr)
    {
        return tree[u].sm1;
    }
    push_down(u,l,r);
    int mid=(l+r)>>1;
    if(tr<=mid) return Query2(lz,tl,tr);
    else if(tl>mid) return Query2(rz,tl,tr);
    else
    {
        int t1=Query2(lz,tl,mid);
        int t2=Query2(rz,mid+1,tr);
        int  t=max(t1,t2);
        t1=min(tree[2*u].rm1,mid-tl+1);
        t2=min(tree[2*u+1].lm1,tr-mid);
        t1+=t2;
        return max(t,t1);
    }
}

int main()
{
    int T, n, m;
    cin >> T;
    while(T--)
    {
        cin >> n >> m;
        for(int i=0; i<n; i++)
            scanf("%d",a+i);
        build(1,0,n-1);
        while(m--)
        {
            int op, l, r;
            scanf("%d%d%d",&op,&l,&r);
            op++;
            if(op==1) Update(1,0,n-1,l,r,1);
            else if(op==2) Update(1,0,n-1,l,r,2);
            else if(op==3) Update(1,0,n-1,l,r,3);
            else if(op==4)
            {
                int sum=Query1(1,0,n-1,l,r);
                printf("%d\n",sum);
            }
            else
            {
                int len=Query2(1,0,n-1,l,r);
                printf("%d\n",len);
            }
        }
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值