Count Color poj2777 线段树

题意

有一个长木板，现在往上面在一定区间内刷颜色，后来刷的颜色会掩盖掉前面刷的颜色，问每次一定区间内可以看到多少种颜色。

解题思路

这里使用线段树，因为刷颜色可以看作是区间修改，使用lazy标记区间的颜色种类，下传标记后，当前节点的lazy标记就标记为0，然后使用vis数组来标记颜色（颜色种类很少）。剩下的基本就是线段树的模板了。

代码实现

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=1e5+17;
struct node{int l, r, lazy;
}a[maxn<<2];
int vis[35];//颜色种类
int L, T, O, ans;
void build(int rt, int l, int r)
{a[rt].l=l;a[rt].r=r;a[rt].lazy=1;//默认为第一种颜色if(l==r){return ;}int mid=(l+r)>>1;build(rt<<1, l, mid);build(rt<<1|1, mid+1, r);
}
void down(int k)
{a[k<<1].lazy=a[k].lazy;a[k<<1|1].lazy=a[k].lazy;a[k].lazy=0;//表示他的节点下面可能是两种不同的颜色
}
void update(int rt, int L, int R, int x)
{if(L<=a[rt].l && a[rt].r<=R){a[rt].lazy=x;return ; }int mid=(a[rt].l+a[rt].r)>>1;if(a[rt].lazy !=0 ) //记得一定要下传标记 down(rt); if(L<=mid) update(rt<<1, L, R, x);if(R>mid) update(rt<<1|1, L, R, x);
}
void query(int rt, int L, int R)
{if(a[rt].lazy!=0) //如果不为零就可以进行判断，因为下面的也是这种颜色{if(!vis[a[rt].lazy])// 看是否之前标记过{ans++; //没有标记就加一vis[a[rt].lazy]=1; //标记}return ;}int mid=(a[rt].l+a[rt].r)>>1;if(a[rt].lazy!=0)down(rt);if(L<=mid) query(rt<<1, L, R);if(R>mid) query(rt<<1|1, L, R);
}
int main()
{int l, r, c;char s[4];while(scanf("%d%d%d", &L, &T, &O)!=EOF){build(1, 1, L);for(int i=1; i<=O; i++){scanf("%s", s);if(s[0]=='C'){scanf("%d%d%d", &l, &r, &c);if(l > r){swap(l, r);}update(1, l, r, c);}else {scanf("%d%d", &l, &r);if(l > r) {swap(l, r); }memset(vis, 0, sizeof(vis));ans=0;query(1, l, r);printf("%d\n", ans);}}   }return 0;
}