POJ2777Count Color

线段树+位操作

1.   题意：有一块长为L的长坂，给各个小块涂色，数据给出的要么是A到B上颜色C，要么要求输出A到B有多少种颜色.

2.   其实这道题一看就知道是线段树，而且最多30种颜色，不也难想出用位操作，不过后面一直卡在颜色的更新上，因为如果颜色要更新到叶子节点会超时，如果更新到刚好匹配的区间，对于后来更新的颜色又会有所冲突，要怎么解决呢？？？

3.   其实方法就是给每个节点加上一个标记cover，如果更新到一个匹配的区间，就把cover赋值为true，这样做有3个用处

(1). 在进行输出dfs时，如果找到的节点的cover是true时，就可以直接返回它的colorstate，因为它就就只有一个颜色

(2). 在后来更新颜色时，如果找到一个点cover为true，就需要先把自己变为false，两个子节点赋值为true，而且把状态赋值给子节点，因为它的子节点的颜色时不正确的，真正的颜色在遇到的第一个cover为true时就被覆盖了，所以要更新子节点，之后在进行正常遍历

(3). 不需要进行buildtree操作.

 

附代码

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>

#define L 101000

using namespace std;

int state[L<<2];
int n,t,o;
int mi[33];
bool f[L<<2];
int getcolor(int stat)
{
    int ret=0;
    while(stat)
    {
        stat&=stat-1;
        ++ret;
    }
    return ret;
}

int DFS(int pos,int st,int en,int ll,int rr)
{
    if(f[pos])
    {
        return state[pos];
    }
    if(st==ll && rr==en)
    {
        return state[pos];
    }
    int mid=(st+en)>>1;
    if(ll > mid)
    {
        return DFS(pos<<1|1,mid+1,en,ll,rr);
    }
    if(rr <= mid)
    {
        return DFS(pos<<1,st,mid,ll,rr);
    }
    return  DFS(pos<<1,st,mid,ll,mid) | DFS(pos<<1|1,mid+1,en,mid+1,rr);
}
void insert(int pos,int st,int en,int ll,int rr,int color)
{
    if(st==ll && en ==rr)
    {
        state[pos]=mi[color];
        f[pos]=true;
        return;
    }
    int mid=(st+en)>>1;

    if(f[pos])
    {
        state[pos<<1]=state[pos];
        state[pos<<1|1]=state[pos];
        f[pos<<1]=true;
        f[pos<<1|1]=true;
    }
    if(ll>mid)
    {
        insert(pos<<1|1,mid+1,en,ll,rr,color);
    }
    else if(rr<=mid)
    {
        insert(pos<<1,st,mid,ll,rr,color);
    }
    else
    {
        insert(pos<<1|1,mid+1,en,mid+1,rr,color);
        insert(pos<<1,st,mid,ll,mid,color);
    }

    state[pos]=state[pos<<1]|state[pos<<1|1];
    f[pos]=false;
}
int main()
{
    //freopen("input.txt","r",stdin);
    mi[1]=1;
    for(int i=2;i<32;++i)
    {
        mi[i]=mi[i-1]<<1;
    }
    memset(f,true,sizeof(f));
    scanf("%d%d%d",&n,&t,&o);
    int nn=n<<2;

    state[1]=1;
    int X,Y,C;
    char P;
    for(int i=1;i<=o;++i)
    {
          scanf(" %c ",&P);
          if(P =='C')
          {
              scanf("%d %d %d ",&X,&Y,&C);
              if(X>Y) swap(X,Y);
              insert(1,1,n,X,Y,C);
          }
          else
          {
              scanf("%d %d ",&X,&Y);
              if(X>Y) swap(X,Y);
              int tt=DFS(1,1,n,X,Y);
              printf("%d\n",getcolor(tt));
          }
    }
    return 0;
}