【NOIP2016提高A组模拟7.15】颜料大乱斗 题解+代码

Description

Input

Output

对于每个询问，输出答案

Sample Input

2 2 4
C 1 1 2
P 1 2
C 2 2 2
P 1 2

Sample Output

2
1

Solution

区间修改与查询：线段树。
开三十棵线段树或者一棵线段树维护三十个东西还有lazy标记，水题，裸题

另外，白色是1，题目坑人

Code

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define N 101000
#define clear(a) memset(a,0,sizeof(a))
using namespace std;
int n,c,m,bz[31];
struct note{
    int a[31],l;
};
note t[N*5];
void down(int v,int i,int j)
{
    t[v*2].l=t[v*2+1].l=t[v].l;clear(t[v*2].a);clear(t[v*2+1].a);
    int m=(i+j)/2;t[v*2].a[t[v].l]=(m-i+1);t[v*2+1].a[t[v].l]=(j-m);t[v].l=0;
}
void change(int v,int i,int j,int x,int y,int z)
{
    if(i==x&&j==y) {t[v].l=z;clear(t[v].a);t[v].a[z]=j-i+1;return;}
    int m=(i+j)/2;if(t[v].l!=0)down(v,i,j);
    if(y<=m) change(v*2,i,m,x,y,z);
    else if(x>m) change(v*2+1,m+1,j,x,y,z);
         else change(v*2,i,m,x,m,z),change(v*2+1,m+1,j,m+1,y,z);
    fo(k,1,c) t[v].a[k]=t[v*2].a[k]+t[v*2+1].a[k];
}
void find(int v,int i,int j,int x,int y)
{
    if(i==x&&j==y)
    {
        fo(k,1,c)
        if (t[v].a[k]!=0) 
        {
            if(bz[k]==0) bz[0]++;
            bz[k]++;
        }
        return;
    }
    if(t[v].l!=0)down(v,i,j);int m=(i+j)/2;
    if(y<=m) find(v*2,i,m,x,y);
    else if(x>m) find(v*2+1,m+1,j,x,y);
         else find(v*2,i,m,x,m),find(v*2+1,m+1,j,m+1,y);
}
int main()
{
    scanf("%d%d%d\n",&n,&c,&m);
    change(1,1,n,1,n,1);
    fo(i,1,m)
    {
        char ch;int x,y;scanf("%c %d %d",&ch,&x,&y);
        if(x>y) swap(x,y);
        if(ch=='C')
        {
            int z;scanf("%d\n",&z);
            change(1,1,n,x,y,z);
        }
        else
        {
            scanf("\n");
            clear(bz);find(1,1,n,x,y);
            printf("%d\n",bz[0]);
        }
    }
}