多校10-1002 线段树

这道题没想到用一个小小的剪枝竟然过了，看来数据不是灰常强大。。不过过的还是蛮辛苦的，主要是这道题有0，如果说l=0的话，那好办，把l,r都+1就是了。问题是颜色也可以等于0.而这个问题很容易被人忽视，从而掉进陷阱。果然，我开始就忽视掉了，原因还是经验不足。后来才发现一个问题，就是成段更新的时候，延迟标记初始值等于0，而color可以为0。这无疑就冲突了。因此要把延迟数组初始化为-1，最后终于AC了，坑爹呀。。。。。至于内个剪枝，so easy。控制一下最大颜色和最小颜色就是了。。。速度还可以，不到1s。附代码：

#include <iostream>

using namespace std;

#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define clear(a) memset(a,0,sizeof(a))
#define dmax(a,b) (a>b)?a:b
#define dmin(a,b) (a<b)?a:b

const int N=100100;

int smin[N<<3],smax[N<<3],col[N<<3],sum[N<<3];

void init()
{
    clear(smin);
    clear(smax);
    memset(col,-1,sizeof(col));
    clear(sum);
}

void pushup(int rt)
{
    int a,b;
    a=rt<<1;
    b=rt<<1|1;
    smax[rt]=dmax(smax[a],smax[b]);
    smin[rt]=dmin(smin[a],smin[b]);
    if (sum[a]==sum[b]&&sum[a]>=0)
        sum[rt]=sum[a];
    else sum[rt]=-1;
}

void pushdown(int l,int r,int rt)
{
    if (col[rt]>=0)
    {
        int k=col[rt];
        col[rt<<1]=col[rt<<1|1]=k;
        smax[rt<<1]=smax[rt<<1|1]=k;
        smin[rt<<1]=smin[rt<<1|1]=k;
        sum[rt<<1]=sum[rt<<1|1]=k;
        col[rt]=-1;
    }
}

void build(int l,int r,int rt)
{
    if (l==r)
    {
        scanf("%d",&sum[rt]);
        smin[rt]=smax[rt]=sum[rt];
        col[rt]=-1;
        return;
    } 
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}

void update(int L,int R,int z,int l,int r,int rt)
{
    pushdown(l,r,rt);
    if (L<=l&&r<=R)
    {
        smin[rt]=z;
        smax[rt]=z;
        col[rt]=z;
        sum[rt]=z;
        return;
    }
    int m=(l+r)>>1;
    if (L<=m) update(L,R,z,lson);
    if (R>m) update(L,R,z,rson);
    pushup(rt);
}

int query(int L,int R,int z,int l,int r,int rt)
{
    pushdown(l,r,rt);
    if (L<=l&&r<=R)
    {
        if  (sum[rt]==z) 
        {
            return r-l+1;
        }
        if  (sum[rt]>0&&sum[rt]!=z) return 0; 
    }
    int m=(l+r)>>1,res=0;
    if (L<=m)
    {
        if (smax[rt<<1]>=z&&smin[rt<<1]<=z) 
            res+=query(L,R,z,lson);
    }
    if (R>m)
    {
        if (smax[rt<<1|1]>=z&&smin[rt<<1|1]<=z)
            res+=query(L,R,z,rson);
    }
    return res;
}

int main()
{
    int n,m,i;
    while (scanf("%d%d",&n,&m)!=EOF)
    {
        init();
        build(1,n,1);
        while (m--)
        {
            int p,x,y,z;
            scanf("%d%d%d%d",&p,&x,&y,&z);
            x++;y++;
            if (p==1)
            {
                update(x,y,z,1,n,1);
            }
            else
            {
                int s=query(x,y,z,1,n,1);
                printf("%d\n",s);
            }
        }
    }
    return 0;
}