【NOIP2016提高A组模拟7.15】颜料大乱斗

Description
【NOIP2016提高A组模拟7.15】颜料大乱斗_第1张图片

Input
【NOIP2016提高A组模拟7.15】颜料大乱斗_第2张图片
Sample Input
2 2 4
C 1 1 2
P 1 2
C 2 2 2
P 1 2
Output
这里写图片描述
Sample Output
2
1

The Solution

【NOIP2016提高A组模拟7.15】颜料大乱斗_第3张图片

总的来说就是30棵线段树,主要用lazy迁移。第一次打,打得较猥琐。。。

不要说我懒 (╯3╰)。。

#define HownoneHe
#include 
#include 
#include 
#include 
#include 
#define fo(i,a,b) for (int i=a;i<=b;i++)
#define fd(i,a,b) for (int i=a;i>=b;i--)
#define N 100005
using namespace std;
int n,c,ans,lazy[N*4];
bool Tree[35][N*4];
bool bz[35];
void Make(int v,int l,int r)
{
    if(l==r)
    {
        Tree[1][v]=1;
        return;
    }
    int mid=(l+r)>>1;
    Make(v*2,l,mid);
    Make(v*2+1,mid+1,r);
    Tree[1][v]=1;
}
void Update(int x,int z)
{
    fo(i,1,30) Tree[i][x]=0;
    Tree[z][x]=1;
    lazy[x]=z;
}

void Down(int x)
{
    if (!lazy[x]) return;
    Update(x*2,lazy[x]);
    Update(x*2+1,lazy[x]);
    lazy[x]=0;
}

void Change(int v,int l,int r,int x,int y,int z)
{
    if (l>=x&&r<=y)
    {
        Update(v,z);
        return;
    }
    Down(v);
    int mid=(l+r)>>1;
/*  if (y<=mid) Change(v*2,l,mid,x,y,z);
    else if (x>mid) Change(v*2+1,mid+1,r,x,y,z);
    else Change(v*2,l,mid,x,mid,z),Change(v*2+1,mid+1,r,mid+1,y,z);*/
    if (x<=mid) Change(v*2,l,mid,x,y,z);
    if (y>mid) Change(v*2+1,mid+1,r,x,y,z);
    fo(i,1,30) Tree[i][v]=Tree[i][v*2]||Tree[i][v*2+1];
}

void Find(int v,int l,int r,int x,int y)
{
    if (l>=x&&r<=y)
    {
        fo(i,1,30) if (!bz[i] && Tree[i][v]) ans++,bz[i]=true;
        return;
    }
    Down(v);
    int mid=(l+r)>>1;
/*  if (y<=mid) Find(v*2,l,mid,x,y);
    else if (x>mid) Find(v*2+1,mid+1,r,x,y);
    else Find(v*2,l,mid,x,mid),Find(v*2+1,mid+1,r,mid+1,y);*/
    if (x<=mid) Find(v*2,l,mid,x,y);
    if (y>mid) Find(v*2+1,mid+1,r,x,y);
}
int main()
{
    freopen("data.in","r",stdin);
    freopen("data.out","w",stdout);
    char ch;
    int m,x,y,z;
    scanf("%d%d%d\n",&n,&c,&m);
    Make(1,1,n);
    while(m--)
    {
        scanf("%c%d%d",&ch,&x,&y);
        if(x>y) swap(x,y);
        if(ch=='C')
        {
            scanf("%d\n",&z);
            Change(1,1,n,x,y,z);
        }
        else
        {
            scanf("\n");
            ans=0;
            memset(bz,0,sizeof(bz));
            Find(1,1,n,x,y);
            printf("%d\n",ans);
        }
    }
    return 0;
}

你可能感兴趣的:(数据结构,线段树)