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

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

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

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

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<algorithm>
 4 using namespace std;
 5 #define ll long long
 6 struct data{
 7     int l,r;
 8     ll add;
 9     ll sum;
10 }data[400005];
11 
12 void build(int l,int r,int k){
13     data[k].l=l;
14     data[k].r=r;
15     data[k].add=0;
16     data[k].sum=0;
17     if(l==r){
18         return ;
19     }
20     int mid=(data[k].l+data[k].r) >> 1;
21     build(l,mid,k*2);
22     build(mid+1,r,k*2+1);
23 }
24 
25 void pushdown(int k){
26     if(data[k].add==0) return ;
27     data[k*2].sum |= data[k].add;
28     data[k*2+1].sum |= data[k].add;
29     data[k*2].add |= data[k].add;
30     data[k*2+1].add |=data[k].add;
31     data[k].add=0;
32 }
33 
34 void update(int l,int r,int x,int k){
35     if(data[k].l==l&&data[k].r==r){
36         data[k].sum|=(ll)(1LL<<x);
37         data[k].add|=(ll)(1LL<<x);
38         return ;
39     }
40     pushdown(k);
41     int mid=(data[k].l + data[k].r)/2;
42     if(r <= mid){
43         update(l,r,x,k*2);
44     }else if(l > mid){
45         update(l,r,x,k*2+1);
46     }else{
47         update(l,mid,x,k*2);
48         update(mid+1,r,x,k*2+1);
49     }
50     data[k].sum=data[k*2].sum | data[k*2+1].sum;
51 }
52 
53 
54 ll query(int l,int r,int k){
55     if(data[k].l==l&&data[k].r==r){
56         return data[k].sum;
57     }    
58     pushdown(k);
59     int mid=(data[k].l + data[k].r) / 2;
60     if(r <= mid){
61         return query(l,r,k*2);
62     }else if(l > mid){
63         return query(l,r,k*2+1);
64     }else{
65         return query(l,mid,k*2) | query(mid+1,r,k*2+1);
66     }
67 }
68 
69 int main(){
70     int n,m;
71     scanf("%d %d",&n,&m);
72     build(1,n,1);
73     while(m--){
74         int flag;
75         scanf("%d",&flag);
76         if(flag==1){
77             int x,y,k;
78             scanf("%d%d%d",&x,&y,&k);
79             update(x,y,k,1);
80         }else{
81             int x,y;
82             scanf("%d%d",&x,&y);
83             int ans = query(x,y,1);
84             int cnt = 0;
85             while(ans){
86                 if(ans&1)
87                     cnt++;
88                 ans = ans >> 1;
89             }
90             cout<<cnt<<endl;
91         }
92     }
93     return 0;
94 }

View Code