2018湘潭大学程序设计竞赛【H】

题目链接：https://www.nowcoder.com/acm/contest/105/H

题意：两个操作，一个在[l,r]区间放颜色为c的球，一个统计在[l,r]里有多少不同颜色的球。

题解：哎，线段树嘛。板子嘛、把求和操作改一改嘛。当时没用lazy操作T了嘛。谢谢大佬的板子。小萌新知道lazy是个啥玩意儿了。emmm..最后统计一下1的个数就是答案啦。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
#define ll long long
struct data{
    int l,r;
    ll add;
    ll sum;
}data[400005];

void build(int l,int r,int k){
    data[k].l=l;
    data[k].r=r;
    data[k].add=0;
    data[k].sum=0;
    if(l==r){
        return ;
    }
    int mid=(data[k].l+data[k].r) >> 1;
    build(l,mid,k*2);
    build(mid+1,r,k*2+1);
}

void pushdown(int k){
    if(data[k].add==0) return ;
    data[k*2].sum |= data[k].add;
    data[k*2+1].sum |= data[k].add;
    data[k*2].add |= data[k].add;
    data[k*2+1].add |=data[k].add;
    data[k].add=0;
}

void update(int l,int r,int x,int k){
    if(data[k].l==l&&data[k].r==r){
        data[k].sum|=(ll)(1LL<<x);
        data[k].add|=(ll)(1LL<<x);
        return ;
    }
    pushdown(k);
    int mid=(data[k].l + data[k].r)/2;
    if(r <= mid){
        update(l,r,x,k*2);
    }else if(l > mid){
        update(l,r,x,k*2+1);
    }else{
        update(l,mid,x,k*2);
        update(mid+1,r,x,k*2+1);
    }
    data[k].sum=data[k*2].sum | data[k*2+1].sum;
}


ll query(int l,int r,int k){
    if(data[k].l==l&&data[k].r==r){
        return data[k].sum;
    }    
    pushdown(k);
    int mid=(data[k].l + data[k].r) / 2;
    if(r <= mid){
        return query(l,r,k*2);
    }else if(l > mid){
        return query(l,r,k*2+1);
    }else{
        return query(l,mid,k*2) | query(mid+1,r,k*2+1);
    }
}

int main(){
    int n,m;
    scanf("%d %d",&n,&m);
    build(1,n,1);
    while(m--){
        int flag;
        scanf("%d",&flag);
        if(flag==1){
            int x,y,k;
            scanf("%d%d%d",&x,&y,&k);
            update(x,y,k,1);
        }else{
            int x,y;
            scanf("%d%d",&x,&y);
            int ans = query(x,y,1);
            int cnt = 0;
            while(ans){
                if(ans&1)
                    cnt++;
                ans = ans >> 1;
            }
            cout<<cnt<<endl;
        }
    }
    return 0;
}