首页 > 代码库 > POJ 2777 Count Color【线段树】

POJ 2777 Count Color【线段树】

题目大意:要求完成以下两个操作:1.将一个区间刷上一种颜色2.询问一段区间上有多少种颜色

思路:这两个操作线段树都可以很迅速的完成,具体做法是:线段树上每个节点存这个线段上的颜色数量,由于颜色数很少,因此可以用二进制存颜色,如果二进制的第N位是1,则该区间存在颜色N,因此一个节点等于其两个子节点颜色的或。最后一个问题就是修改操作是对区间修改,因此需要用lazy_tag的思想(虽然这里不完全是),不然一个一个插节点会TLE

#include<cstdio>

#include<string.h>

#include<iostream>

using namespace std;

int tree[500000]={0};

bool lazy[500000]={0};

void swap(int &x,int &y){int temp=x;x=y;y=temp;}

int search(int node,int l,int r,int lq,int rq)

{

    int mid=(l+r)>>1,temp=0;

    if (lazy[node]==true){return tree[node];}

    if (lq<=l && r<=rq){return tree[node];}

    else

    {

        if (lazy[node]==1){lazy[node<<1]=1;tree[node<<1]=tree[node];

        tree[(node<<1)+1]=tree[node];lazy[(node<<1)+1]=1;}

        if (lq<=mid)temp=search(node<<1,l,mid,lq,rq);

        if (mid<rq)temp=temp | search((node<<1)+1,mid+1,r,lq,rq);

    }

    return temp;

}

void insert(int node,int l,int r,int lq,int rq,int color)

{

    int mid=(l+r)>>1;

    if (lq<=l && r<=rq){tree[node]=1<<(color-1);lazy[node]=true;return;}

    else

    {

        if (lazy[node]==1)

        {

            lazy[node]=0;lazy[node<<1]=1;

            tree[node<<1]=tree[node];tree[(node<<1)+1]=tree[node];

            lazy[(node<<1)+1]=1;

        }

        if (lq<=mid)insert(node<<1,l,mid,lq,rq,color);

        if (rq>mid)insert((node<<1)+1,mid+1,r,lq,rq,color);

    }

    tree[node]=tree[node<<1] | tree[(node<<1)+1];

}

int f(int u)

{

    int t=0;while(u!=0){if ((u & 1)==1)t++;u>>=1;}

    return t;

}

int main()

{

    int t,o,l,left,right,color;

    char ch[2];

    scanf("%d%d%d",&l,&t,&o);

    for(int i=1;i<=4*l;i++)tree[i]=1;

    for(int v=1;v<=o;v++)

    {

        scanf("%s",ch);

        if (ch[0]==‘C‘)

        {

            scanf("%d%d%d",&left,&right,&color);

                     if (right<left)swap(right,left);

            insert(1,1,l,left,right,color);

        }

        else

        {

            scanf("%d%d",&left,&right);

                     if (right<left)swap(right,left);

            int u=search(1,1,l,left,right);

            printf("%d\n",f(u));

        }

    }

    return 0;

}

POJ 2777 Count Color【线段树】