2018-湘潭大学-H统计颜色

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/ACMER_2333/article/details/80166040

题目描述
n个桶按顺序排列，我们用1~n给桶标号。有两种操作：
1 l r c 区间[l,r]中的每个桶中都放入一个颜色为c的球 (1≤l,r ≤n,l≤r,0≤c≤60)
2 l r   查询区间[l,r]的桶中有多少种不同颜色的球     (1≤l,r ≤n,l≤r)
输入描述:
有多组数据，对于每组数据：
第一行有两个整数n,m(1≤n,m≤100000)
接下来m行，代表m个操作，格式如题目所示。
输出描述:
对于每个2号操作，输出一个整数，表示查询的结果。
示例1
输入
10 10
1 1 2 0
1 3 4 1
2 1 4
1 5 6 2
2 1 6
1 7 8 1
2 3 8
1 8 10 3
2 1 10
2 3 8
输出
2
3
2
4
3
分析：线段树的裸题，这道题主要用到了按位或运算（|），目的是为了统计不同的颜色个数。
代码如下：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <map>
#include <string>
#include <vector>
#include <queue>
#include <cmath>
#include <set>
#include <vector>
using namespace std;
const int Maxn=1e5+10;
typedef long long ll;
struct Node
{
    ll left;
    ll right;
    ll sum;     ///统计不同颜色的个数
    ll lazy;   ///懒惰标记
    ll mid()
    {
        return (left+right)/2;
    }
}node[Maxn*4];
void buildTree(ll i,ll l,ll r)
{
    node[i].left=l;
    node[i].right=r;
    node[i].sum=0;
    node[i].lazy=0;
    if(node[i].left==node[i].right)
        return ;
    ll m=node[i].mid();
    buildTree(i<<1,l,m);
    buildTree(i<<1|1,m+1,r);
}
void push_down(ll i)
{
    if(node[i].lazy)
    {
        node[i<<1|1].sum |=node[i].lazy;
        node[i<<1].sum   |=node[i].lazy;
        node[i<<1|1].lazy|=node[i].lazy;
        node[i<<1].lazy  |=node[i].lazy;
        node[i].lazy=0;
    }
}
void push_up(ll i)
{
    node[i].sum=node[i<<1].sum|node[i<<1|1].sum;
}
void add(ll i,ll l,ll r,ll c)
{
    if(node[i].left==l&&node[i].right==r)
    {
        node[i].sum|=c;
        node[i].lazy|=c;
        return ;
    }
    push_down(i);
    ll m=node[i].mid();
    if(m>=r)
        add(i<<1,l,r,c);
    else if(m<l)
        add(i<<1|1,l,r,c);
    else
    {
        add(i<<1,l,m,c);
        add(i<<1|1,m+1,r,c);
    }
    push_up(i);
}
ll qury(ll i,ll l,ll r)
{
    if(node[i].left==l&&node[i].right==r)
        return node[i].sum;
    push_down(i);
    ll m=node[i].mid();
    if(m>=r)
        return qury(i<<1,l,r);
    else if(l>m)
        return qury(i<<1|1,l,r);
    else
        return qury(i<<1,l,m)|qury(i<<1|1,m+1,r);
}
ll find1(ll x)   ///统计颜色的个数
{
    ll sum=0;
    while(x>0)
    {
        if(x%2)
            sum++;
        x/=2;
    }
    return sum;
}
int main()
{
    //cout<<(1<<3)<<endl;
    ll n,m;
    while(~scanf("%lld%lld",&n,&m))
    {
        buildTree(1,1,n);
        for(ll i=1;i<=m;i++)
        {
            ll ope;
            ll l,r,c;
            scanf("%lld%lld%lld",&ope,&l,&r);
            if(ope==1)
            {
                scanf("%lld",&c);
                add(1,l,r,(1ll<<c));
            }
            else
                printf("%lld\n",find1(qury(1,l,r)));
        }
    }
    return 0;
}