线段树的模板题。
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <queue>
#include <algorithm>
using namespace std;
struct N
{
int l,r,col;
bool MarkVisit;
} st[400100];
void init(int l,int r,int site)
{
st[site].l = l;
st[site].r = r;
st[site].col = 1;
st[site].MarkVisit = true;//标记为单色
if(l == r)
return ;
int mid = (l+r)>>1;
init(l,mid,site<<1);
init(mid+1,r,site<<1|1);
}
void updata(int l,int r,int col,int site)
{
if(l == st[site].l && r == st[site].r)
{
st[site].col = col;
st[site].MarkVisit = true;
return ;
}
if(st[site].MarkVisit && st[site].col == col)
return ;
int mid = (st[site].l + st[site].r)>>1;
if(st[site].MarkVisit)
{
updata(st[site].l,mid,st[site].col,site<<1);
updata(mid+1,st[site].r,st[site].col,site<<1|1);
}
st[site].MarkVisit = false;
if(r <= mid)
{
updata(l,r,col,site<<1);
}
else if(mid < l)
{
updata(l,r,col,site<<1|1);
}
else
{
updata(l,mid,col,site<<1);
updata(mid+1,r,col,site<<1|1);
}
}
bool MarkColor[40];
int query(int l,int r,int site)
{
if(st[site].l <= l && r <= st[site].r && st[site].MarkVisit == true)
{
if(MarkColor[st[site].col] == false)
{
MarkColor[st[site].col] = true;
return 1;
}
return 0;
}
int mid = (st[site].l + st[site].r)>>1;
if(r <= mid)
{
return query(l,r,site<<1);
}
if(mid < l)
{
return query(l,r,site<<1|1);
}
return query(l,mid,site<<1) + query(mid+1,r,site<<1|1);
}
int main()
{
int len,t,o;
int l,r,col;
char op;
while(scanf("%d %d %d",&len,&t,&o) != EOF)
{
init(1,len,1);//初始化
while(o--)
{
scanf("%*c%c",&op);
if(op == 'C')
{
scanf("%d %d %d",&l,&r,&col);
updata(l,r,col,1);
}
else
{
scanf("%d %d",&l,&r);
memset(MarkColor,false,(t+2)*sizeof(bool));
printf("%d\n",query(l,r,1));
}
}
}
return 0;
}