線段樹 經典的題目，以前曾經做過一遍，現在為了練手在做一次，剛學了splay樹，反倒是加深了對線段樹的理解，就是那個延遲標記（也就是懶操作）。雖然線段樹已經寫過多次，但是這題仍然不能1A,Query函數中有個地方應該是mid=(ST[i].l+ST[i].r)>>1寫成了(l+r)>>1,導致wa了幾次，今后要注意啊。
#include<iostream>
using namespace std;
int const maxn=100010;
int n,t,q;
struct node
{
int l,r;
int col;//用位來存儲顏色
int cover;//延遲標記
}ST[maxn*4];
void Build(int l,int r,int i)
{
ST[i].l=l;
ST[i].r=r;
ST[i].col=0;
ST[i].cover=0;
if(l==r)
return;
int mid=(l+r)>>1;
Build(l,mid,i*2);
Build(mid+1,r,i*2+1);
}
void push_down(int i)
{
ST[i*2].col=ST[i].col;
ST[i*2+1].col=ST[i].col;
ST[i].cover=0;
ST[i*2].cover=1;
ST[i*2+1].cover=1;
}
void insert(int l,int r,int col,int i)
{
if(ST[i].l==l&&ST[i].r==r)
{
ST[i].cover=1;
ST[i].col=(1<<(col-1));
return ;
}
if(ST[i].cover)//如果當前區間有效，下沿延遲標記
push_down(i);
int mid=(ST[i].l+ST[i].r)>>1;
if(r<=mid)
insert(l,r,col,i*2);
else if(l>mid)
insert(l,r,col,i*2+1);
else
{
insert(l,mid,col,i*2);
insert(mid+1,r,col,i*2+1);
}
ST[i].col=ST[i*2].col|ST[i*2+1].col;
}
int fun(int num)//檢查最后返回的整數中有多少顏色
{
int ans=0;
int i;
for(i=0;i<t;i++)
if(num&(1<<i))
ans++;
return ans;
}
int Que(int l,int r,int i)
{
if( (ST[i].l==l&&ST[i].r==r)||ST[i].cover==1)
return ST[i].col;
int mid=(ST[i].l+ST[i].r)>>1;
if(r<=mid)
return Que(l,r,i*2);
else if(l>mid)
return Que(l,r,i*2+1);
else
return Que(l,mid,i*2)|Que(mid+1,r,i*2+1);
}
int main()
{
while(scanf("%d%d%d",&n,&t,&q)!=EOF)
{
Build(1,n,1);
ST[1].cover=1;
ST[1].col=1;
char op[20];
int a,b,c;
for(int i=1;i<=q;i++)
{
scanf("%s",op);
if(op[0]=='C')
{
scanf("%d%d%d",&a,&b,&c);
if(a>b)
swap(a,b);
insert(a,b,c,1);
}
else
{
scanf("%d%d",&a,&b);
if(a>b)
swap(a,b);
printf("%d\n",fun(Que(a,b,1)));
}
}
}
return 0;
}