PKU 2777 Count Color （线段树区间更新）

来源：互联网发布：福山大学知乎编辑：程序博客网时间：2024/06/05 15:30

题意：给你三个数：L (1 <= L <= 100000), T (1 <= T <= 30) and O (1 <= O <= 100000)，表示有一长度为L的板（1~L），

有T种颜色（1~T），然后有O个操作，初始板1~L的颜色为1，"C A B C"表示在区间A，B图上C颜色， "P A B" 表示询问

A，B区间有几种不同的颜色。

#include <stdio.h>  #include <iostream>  #include <algorithm>  #include <string.h>  #include <math.h>  #define M 100000  #define LL long long  using namespace std;  struct Tree{int l,r,c;//c代表颜色，若区间（l，r）的颜色不止一种则c=-1；}tree[M*3];int color[35];//在查询时，若[A,B]区间内出现过某种颜色k,则color[k]=true;void build(int id,int l,int r){tree[id].l=l;tree[id].r=r;tree[id].c=1;//初始颜色为1if(l==r) return ;build(id*2,l,(l+r)/2);build(id*2+1,(l+r)/2+1,r);}void update(int id,int l,int r,int co){if(tree[id].l>=l&&tree[id].r<=r)//（l，r）完全覆盖id节点{tree[id].c=co;return;}if(tree[id].c>0)//一直wa在这里{tree[id*2].c=tree[id*2+1].c=tree[id].c;tree[id].c=-1;}tree[id].c=-1;//没有完全覆盖int mid=(tree[id].l+tree[id].r)/2;if(mid>=r)update(2*id,l,r,co);else if(mid<l)update(2*id+1,l,r,co);else {update(2*id,l,mid,co);update(2*id+1,mid+1,r,co);}}void P(int id,int l,int r){if(tree[id].c>0){color[tree[id].c]=true;return;}if(tree[id].l==tree[id].r) return;int mid=(tree[id].l+tree[id].r)/2;if(mid>=r)P(id*2,l,r);else if(mid<l)P(id*2+1,l,r);else {P(id*2,l,mid);P(id*2+1,mid+1,r);}}int Ans(int t){int ans=0;for(int i=1;i<=t;i++)if(color[i])ans++;return ans;}int n,t,m;int main(){scanf("%d%d%d",&n,&t,&m);char s[2];int a,c,b;build(1,1,n);while(m--){scanf("%s",s);if(s[0]=='C'){scanf("%d%d%d",&a,&b,&c);if(a>b) swap(a,b);update(1,a,b,c);}else {scanf("%d%d",&a,&b);if(a>b) swap(a,b);memset(color,false,sizeof(color));P(1,a,b);printf("%d\n",Ans(t));}}return 0;}

0 0