Paint The Wall hdu 4391 (线段树)

Paint The Wall

 HDU - 4391 

解题思路：涉及到线段树的区间赋值和一个小的优化，维护区间颜色的最小标号和最大标号。如果要查询的颜色不在这个区间直接返回就可以了。还有分块的做法。

#include<iostream>
#include<cstdio>
#include<cmath>
using namespace std;
#define LL long long
#define sca(x) scanf("%d",&x)

const int N = 1e5+5;
int a[N];
struct node
{
    int mn,mm,col;
}t[N*4];

void push_up(int rt)
{
    if(t[rt<<1].col!=t[rt<<1|1].col)t[rt].col=-1;
    else t[rt].col=t[rt<<1].col;
    t[rt].mn=min(t[rt<<1].mn,t[rt<<1|1].mn);
    t[rt].mm=max(t[rt<<1].mm,t[rt<<1|1].mm);
}

void push_down(int rt)
{
    if(t[rt].col!=-1)
    {
        t[rt<<1].col=t[rt<<1|1].col=t[rt].col;
        t[rt<<1].mn=t[rt<<1|1].mn=t[rt].col;
        t[rt<<1].mm=t[rt<<1|1].mm=t[rt].col;
    }
}

void build(int rt,int l,int r)
{
    if(l==r)
    {
        t[rt].col=t[rt].mm=t[rt].mn=a[l];
        return ;
    }
    int m=(l+r)>>1;
    build(rt<<1,l,m);
    build(rt<<1|1,m+1,r);
    push_up(rt);
}

void upd(int rt,int l,int r,int ql,int qr,int k)
{
    if(l>=ql && r<=qr)
    {
        t[rt].col=k;
        t[rt].mm=t[rt].mn=k;
        return ;
    }
    push_down(rt);
    int m=(l+r)>>1;
    if(ql<=m) upd(rt<<1,l,m,ql,qr,k);
    if(qr>m) upd(rt<<1|1,m+1,r,ql,qr,k);
    push_up(rt);
}

int ask(int rt,int l,int r,int ql,int qr,int k)
{
    if(k<t[rt].mn||k>t[rt].mm)return 0;
    if(l>=ql && r<=qr)
    {
        if(t[rt].col==k)return r-l+1;
    }
    if(l==r)
    {
        if(t[rt].col==k)return 1;
        else return 0;
    }
    int m=(l+r)>>1;
    push_down(rt);
    int ans=0;
    if(ql<=m) ans+=ask(rt<<1,l,m,ql,qr,k);
    if(qr>m) ans+=ask(rt<<1|1,m+1,r,ql,qr,k);
    return ans;
}

int main()
{
    int n,m;
    while(cin>>n>>m)
    {
        for(int i=1;i<=n;i++)  sca(a[i]);
        build(1,1,n);
        for(int i=1;i<=m;i++)
        {
            int op,l,r,z;
            scanf("%d%d%d%d",&op,&l,&r,&z);
            l++,r++;
            if(op==1)upd(1,1,n,l,r,z);
            else printf("%d\n",ask(1,1,n,l,r,z));
        }
    }
}