[线段树] Jzoj P1214 项链工厂

Description

　　T 公司是一家专门生产彩色珠子项链的公司，其生产的项链设计新颖、款式多样、价格适中，广受青年人的喜爱。最近T 公司打算推出一款项链自助生产系统，使用该系统顾客可以自行设计心目中的美丽项链。
　　该项链自助生产系统包括硬件系统与软件系统，软件系统与用户进行交互并控制硬件系统，硬件系统接受软件系统的命令生产指定的项链。该系统的硬件系统已经完成，而软件系统尚未开发，T 公司的人找到了正在参加全国信息学竞赛的你，你能帮助T 公司编写一个软件模拟系统吗？
　　一条项链包含N 个珠子，每个珠子的颜色是1, 2, …, c 中的一种。项链被固定在一个平板上，平板的某个位置被标记位置1，按顺时针方向其他位置被记为2,3,…,N。 　　
　　你将要编写的软件系统应支持如下命令：
　　1.R k(0 < K < N) 意为Rotate k。将项链在平板上顺时针旋转k 个位置, 即原来处于位置1 的珠子将转至位置k+1，处于位置2 的珠子将转至位置k+2，依次类推。
　　2. F 意为Flip。将平板沿着给定的对称轴翻转，原来处于位置1 的珠子不动，位置2 上的珠子与位置N 上的珠子互换，位置3 上的珠子与位置N-1 上的珠子互换，依次类推。
　　3.S i j(1≤I,j≤N) 意为Swap i , j。将位置i 上的珠子与位置j 上的珠子互换。
　　4.P I j x(1≤I,j≤N,x≤c) 意为Paint i , j , x。将位置i 沿顺时针方向到位置j 的一段染为颜色x。
　　5.C 意为Count。查询当前的项链由多少个“部分”组成，我们称项链中颜色相同的一段为一个“部分”。
　　6.CS i j(1≤I,j≤N) 意为CountSegment i , j。查询从位置i沿顺时针方向到位置j 的一段中有多少个部分组成。

 

Input

　　输入文件第一行包含两个整数N, c，分别表示项链包含的珠子数目以及颜色数目。
第二行包含N 个整数，x1, x2…, xn，表示从位置1 到位置N 的珠子的颜色，1 ≤ xi ≤ c。
第三行包含一个整数Q，表示命令数目。
接下来的Q 行每行一条命令，如上文所述。

Output

　　对于每一个C 和CS 命令，应输出一个整数代表相应的答案。

 

Sample Input

5 3
1 2 3 2 1
4
C
R 2
P 5 5 2
CS 4 1

Sample Output

4
1

 

Data Constraint

 

 

Hint

【数据规模和约定】
　　对于60%的数据，N ≤ 1 000，Q ≤ 1 000；
　　对于100%的数据，N ≤ 500 000，Q ≤ 500 000，c ≤ 1 000。

题解

• 主要是翻转和旋转操作比较奇怪，其他都是线段树基本操作
• 一个是顺时针旋转，如果不存在翻转操作
• 我们显然可以通过记录顺时针旋转的次数来还原当前的一号位置
• 多了一个翻转
• 仔细观察，就是把顺时针变为了逆时针而已
• 所以直接乘个负号，额外加一个标记就行了

代码

#include <cstdio>
#include <iostream>
#include <cstring>
using namespace std;
bool rev;
char ch[5];
int n,m,p;
struct data{int l,r,sum;};
struct tree{int l,r,tag;data v;}t[2000005];
data merge(data a,data b)
{
    data r={a.l,b.r,a.sum+b.sum};
    if (a.r==b.l) r.sum--;
    return r;
}
void calc(int &x,int &y)
{
    if (rev) x=(n*2+2-p-x)%n,y=(n*2+2-p-y)%n,swap(x,y);
    else x=(x-p+n)%n,y=(y-p+n)%n;
    if (x==0) x=n; 
    if (y==0) y=n;
}
void pushdown(int d)
{
    if (!t[d].tag||t[d].l==t[d].r) return;
    int r=t[d].tag; t[d].tag=0;
    t[d*2].tag=t[d*2+1].tag=t[d*2].v.l=t[d*2+1].v.l=t[d*2].v.r=t[d*2+1].v.r=r;
    t[d*2].v.sum=t[d*2+1].v.sum=1;
} 
void build(int d,int l,int r)
{
    t[d].l=l,t[d].r=r;
    if (l==r)
    {
        scanf("%d",&t[d].v.l),t[d].v.r=t[d].v.l,t[d].v.sum=1;
        return;
    }
    int mid=l+r>>1;
    build(d*2,l,mid),build(d*2+1,mid+1,r),t[d].v=merge(t[d*2].v,t[d*2+1].v);
}
data query(int d,int l,int r)
{
    pushdown(d);
    int L=t[d].l,R=t[d].r;
    if (l==L&&r==R) return t[d].v;
    int mid=L+R>>1;
    if (r<=mid) return query(d*2,l,r); else if (l>mid) return query(d*2+1,l,r); else return merge(query(d*2,l,mid),query(d*2+1,mid+1,r));
}
void modify(int d,int l,int r,int x)
{
    pushdown(d);
    int L=t[d].l,R=t[d].r;
    if (l==L&&R==r)
    {
        t[d].v.l=t[d].v.r=t[d].tag=x,t[d].v.sum=1;
        return;
    }
    int mid=L+R>>1;
    if (r<=mid) modify(d*2,l,r,x); else if (l>mid) modify(d*2+1,l,r,x); else modify(d*2,l,mid,x),modify(d*2+1,mid+1,r,x);
    t[d].v=merge(t[d*2].v,t[d*2+1].v);
}
int main()
{
    freopen("data.in","r",stdin);
    scanf("%d%d",&n,&m),build(1,1,n),scanf("%d",&m);
    for (int x,y,z;m;m--)
    {
        scanf("%s",ch);
        if (ch[0]=='R')
        {
            scanf("%d",&x);
            if (rev) p=(p+n-x)%n; else p=(p+x)%n;
        } 
        if (ch[0]=='F') rev^=1;
        if (ch[0]=='S')
        {
            int a,b; data ans;
            scanf("%d%d",&x,&y),calc(x,y); 
            if (x>y) ans=query(1,y,x),a=ans.r,b=ans.l; else ans=query(1,x,y),a=ans.l,b=ans.r;
            modify(1,x,x,b),modify(1,y,y,a);
        }
        if (ch[0]=='P')
        {
            scanf("%d%d%d",&x,&y,&z),calc(x,y);
            if (x<=y) modify(1,x,y,z); else modify(1,x,n,z),modify(1,1,y,z);
        }
        if (ch[0]=='C'&&ch[1]=='S')
        {
            scanf("%d%d",&x,&y),calc(x,y); data ans;
            if (x<=y) ans=query(1,x,y); else ans=merge(query(1,x,n),query(1,1,y));
            printf("%d
",ans.sum);
        }
        if (ch[0]=='C'&&ch[1]!='S')
        {
            data ans=query(1,1,n);
            int r=ans.sum;
            if (ans.l==ans.r) r=max(r-1,1);
            printf("%d
",r);
        }
    }
}