统计颜色（线段树区间修改问题）

题链接：

点击打开链接

题目描述

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

这个题就是裸的线段树，但是我好久没写这个，然后忘记了该怎么用二进制表示不同颜色气球的个数。

然后就是求二进制下有多少个1，手写函数比较好，不偷懒。

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define ll long long
const int maxn=1e5+10;
int n,m,kin,l,r,c;
struct Node{
	int l,r;
	ll lazy,x;
}node[maxn<<2];
void build(int k,int l,int r){
	node[k].l=l;node[k].r=r;node[k].lazy=0;node[k].x=0;
	if(l==r) return ;
	int mid=(l+r)>>1;
	build(k<<1,l,mid);
	build(k<<1|1,mid+1,r);
}
void up(int k,int l,int r,ll c){
	node[k].x|=(1ll<<c);
	if(node[k].l==l&&node[k].r==r){
		node[k].lazy|=(1ll<<c);
		return ;
	}
	int mid=(node[k].l+node[k].r)>>1;
	if(r<=mid) up(k<<1,l,r,c);
	else if(l>mid) up(k<<1|1,l,r,c);
	else{
		up(k<<1,l,mid,c),up(k<<1|1,mid+1,r,c);
	}
}
ll query(int k,int l,int r){
	if(node[k].l==l&&node[k].r==r){
		return node[k].x;
	}
	int mid=(node[k].l+node[k].r)>>1;
	ll le=node[k].lazy;
	if(r<=mid ) return le|query(k<<1,l,r);
	else if(l>mid) return le|query(k<<1|1,l,r);
	else return le|query(k<<1,l,mid)|query(k<<1|1,mid+1,r);
}
int find(ll t){
	int cnt=0;
	while(t){
		cnt++;
		t&=(t-1);
	}
	return cnt;
}
int main(){
	int i,j;
	while(scanf("%d%d",&n,&m)!=EOF){
		build(1,1,n);
		for(i=0;i<m;i++){
			scanf("%d",&kin);
			if(kin==1){
				scanf("%d%d%d",&l,&r,&c);
				up(1,l,r,c);
			}
			else{
				scanf("%d%d",&l,&r);
				ll ans=query(1,l,r);
				printf("%d\n",find(ans));
			}
		}
	}
	return 0;
}