【bzoj1493】项链工厂【线段树】

传送门：http://www.lydsy.com/JudgeOnline/problem.php?id=1493

这题可以用线段树，不过要额外维护左端点颜色、右端点颜色，以及这条线段上颜色有几个“部分”。

下传标记，当且仅当当前线段是纯色的时候（否则它下面一定有更晚染色的线段）。

吐槽一下，我写pushdown的时候把数组名rc写成了lc......

真是人民群众喜闻乐见。

敬请指教，神犇轻喷- -

#include<cstdio>
#include<iostream>
#include<cstdlib>
#define mid (l+r>>1)
using namespace std;
typedef int arr[1001000];
arr lc,rc,same;
struct Data{int l,r,cnt;}data[1001000];//Left color,Right Color,count of Colors.
int tree,tot;
int n,m;
inline void read(int &x){
x=0;
bool f=false;
char ch=getchar();
while((ch<'0'||ch>'9')&&ch!=-1) {
if(ch=='-') f=true;ch=getchar();
}
while(ch>='0'&&ch<='9'){
x=x*10+ch-48;
ch=getchar();
}
if(f) x=-x;
}
inline Data merge(const Data &a,const Data &b){
return (Data){a.l,b.r,a.cnt+b.cnt-(a.r==b.l)};
}
inline void update(int x){
data[x]=merge(data[lc[x]],data[rc[x]]);
if(same[lc[x]]&&same[lc[x]]==same[rc[x]])
same[x]=same[lc[x]];
else same[x]=0;
}
void build(int &x,int l,int r){
x=++tot;
if(l+1<r){
build(lc[x],l,mid);
build(rc[x],mid,r);
update(x);
}
else{
data[x].cnt=1;
read(data[x].l);
data[x].r=data[x].l;
same[x]=data[x].l;
//      same[x]=
}
}
void pushdown(int x){
if(same[x]){
data[lc[x]].l=data[lc[x]].r=
data[rc[x]].l=data[rc[x]].r=same[x];
data[lc[x]].cnt=data[rc[x]].cnt=1;//FUCK！！！！！
same[lc[x]]=same[rc[x]]=same[x];
}//data[lc[x]].cnt=data[lc[x]].cnt=1;
}
void paint(int x,int l,int r,int L,int R,int c){
if(L<=l&&r<=R){
data[x].l=data[x].r=c;
data[x].cnt=1;
same[x]=c;
}
else{
pushdown(x);
if(L<mid) paint(lc[x],l,mid,L,R,c);
if(R>mid) paint(rc[x],mid,r,L,R,c);
update(x);
}
}
Data query(int x,int l,int r,int L,int R){
//  if(!x) exit(-1);
if(L<=l&&r<=R) return data[x];
pushdown(x);//我发现好像只有纯色时 才需要向下更新- -
if(R<=mid) return query(lc[x],l,mid,L,R);
if(L>=mid) return query(rc[x],mid,r,L,R);
return merge(query(lc[x],l,mid,L,R),query(rc[x],mid,r,L,R));
}
int calc(int x,int begin,bool flip){
if(!flip) return ((begin+x-1)%n+n)%n;
return ((begin-x+1)%n+n)%n;
}
int main(){
//freopen("input.in","r",stdin);
int begin=0,q,x,i,j,colorCount;
bool flip=false;
Data tmp1,tmp2;
char cmd[5];
read(n);
read(colorCount);
build(tree,0,n);
read(q);
while(q--){
scanf("%s",cmd);
switch(cmd[0]){
case 'R':read(x);
if(!flip)begin=((begin-x)+n)%n;
else begin=(begin+x)%n;
break;
case 'F':flip^=1;break;
case 'S':
read(i);
read(j);
i=calc(i,begin,flip);
j=calc(j,begin,flip);
tmp1=query(1,0,n,i,i+1);
tmp2=query(1,0,n,j,j+1);
paint(1,0,n,j,j+1,tmp1.l);
paint(1,0,n,i,i+1,tmp2.l);
break;
case 'P':
read(i);
read(j);
read(x);
i=calc(i,begin,flip);
j=calc(j,begin,flip);
//              cout<<"in fact,i="<<i<<",and j="<<j<<"\n";

if(!flip){
if(i>j){
paint(1,0,n,0,j+1,x);
paint(1,0,n,i,n,x);
}
else paint(1,0,n,i,j+1,x);
}
else{
if(i<j){
paint(1,0,n,0,i+1,x);
paint(1,0,n,j,n,x);
}
else paint(1,0,n,j,i+1,x);
}
break;
case 'C':
if(cmd[1]=='S'){
read(i);read(j);
i=calc(i,begin,flip);
j=calc(j,begin,flip);
//                  cout<<"begin="<<begin<<"\n";
//                  cout<<"in fact,i="<<i<<",and j="<<j<<"\n";
if(!flip){
if(i>j){
tmp1=query(1,0,n,0,j+1);
tmp2=query(1,0,n,i,n);
//                          printf("%d,%d,%d,%d\n",tmp1.l,tmp2.r,tmp1.cnt,tmp2.cnt);
printf("%d\n",tmp1.cnt+tmp2.cnt-(tmp1.l==tmp2.r));
}
else printf("%d\n",query(1,0,n,i,j+1).cnt);
}
else{
if(i<j){
tmp1=query(1,0,n,0,i+1);
tmp2=query(1,0,n,j,n);
printf("%d\n",tmp1.cnt+tmp2.cnt-(tmp1.l==tmp2.r));
}
else printf("%d\n",query(1,0,n,j,i+1).cnt);
}
}
else printf("%d\n",max(data[1].cnt-(data[1].l==data[1].r),1));
break;
}
}
return 0;
}