poj 2777(线段树+lazy思想) 小小粉刷匠

http://poj.org/problem?id=2777

题目大意 涂颜色,输入长度,颜色总数,涂颜色次数,初始颜色都为1,然后当输入为C的时候将x到y涂为颜色z,输入为Q的时候输出x到y的颜色总数

很明显的区间线段树,然后加lazy思想记录

lazy操作为了避免查找到每一个子节点区间而费时,将查找到的区间作标记,但查找到这个区间或还要继续像下查找的时候

将此区间的数据传给下面两个区间树

因为这题颜色总类只有30种很少,所以偷了个懒,将判断与记录操作合并到一个结构体上了,然后用类似hash的数组记录颜色总数

code

 #include<cstdio>
 #include<string.h>
 using namespace std;
 struct point
 {
     int l,r;
     int color;
 };
 point tree[*];
 int visit[];
 void build(int i,int left,int right)
 {
     tree[i].l=left;tree[i].r=right;
     tree[i].color=;
     if(left==right)return;
     int mid=(left+right)/;
     build(i*,left,mid);
     build(i*+,mid+,right);
 }
 void update(int i,int left,int right,int val)
 {
     if (left<=tree[i].l&&tree[i].r<=right){tree[i].color=val;return ;}
     if (tree[i].color!=-) //lazy操作
     {
         tree[i*].color=tree[i*+].color=tree[i].color;
         tree[i].color=-;
     }
     if (left<=tree[i*].r) update(i*,left,right,val);
     if (right>=tree[i*+].l) update(i*+,left,right,val);
 }
 void find(int i,int left,int right)
 {
     if (tree[i].color!=-) {visit[tree[i].color]=;return ;}
     if (tree[i].l==tree[i].r) return;
     if (tree[i].color!=-)
     {
         tree[i*].color=tree[i*+].color=tree[i].color;
         tree[i].color=-;
     }
     if (left<=tree[i*].r) find(i*,left,right);
     if (right>=tree[i*+].l) find(i*+,left,right);
 }
 int main()
 {
     int x,y,z,n,m,k;
     char op;
     while (~scanf("%d %d %d",&x,&y,&z))
     {
         build(,,x);
         while (z--)
         {
             scanf(" %c",&op);
             if (op=='C')
             {
                 scanf("%d %d %d",&n,&m,&k);
                 update(,n,m,k);
             }
             else
             {
                 scanf("%d %d",&n,&m);
                 memset(visit,,sizeof(visit));
                 find(,n,m);
                 int sum=;
                 for (int i=;i<=;i++)
                     if (visit[i]==)
                        sum++;
                 printf("%d\n",sum);
             }
         }
     }
     return ;
 }