【数据结构 线段树】JZOJ_1214 【NOI2007】项链工厂

题意

给出一串项链。
编写软件系统支持如下命令：
1.$R$ $k$ 意为$Rotate$ $k$。将项链在平板上顺时针旋转$k$个位置, 即原来处于位置1的珠子将转至位置$k+1$，处于位置2的珠子将转至位置$k+2$，依次类推。
2. $F$ 意为$Flip$。将平板沿着给定的对称轴翻转，原来处于位置1 的珠子不动，位置2 上的珠子与位置$N$上的珠子互换，位置3 上的珠子与位置$N-1$上的珠子互换，依次类推。
3.$Sij$ 意为$Swapi,j$。将位置i 上的珠子与位置j 上的珠子互换。
4.$Pijx$ 意为$Painti,j,x$。将位置i 沿顺时针方向到位置j 的一段染为颜色x。
5.$C$ 意为$Count$。查询当前的项链由多少个“部分”组成，我们称项链中颜色相同的一段为一个“部分”。
6.$CSij$意为$CountSegmenti,j$。查询从位置$i$沿顺时针方向到位置$j$的一段中有多少个部分组成。

思路

除去1、2操作，那么就是一个普通的线段树。
我们发现1、2操作中是直接对整一个环进行操作，所以不会影响其中的顺序，于是可以记录当前环是否翻转或者旋转，在点查询中转换为相应的位置。

代码

#include<cstdio>
#include<algorithm>

struct SegmentTree {
	int l, r, dat, cl, cr, add;
}t[2000001];
int n, c, f, x, q;
int a[500001];
char op[2];

void spread(int p) {
	if (t[p].add) {
		t[p * 2].cl = t[p * 2].cr = t[p * 2].add = t[p].add;
		t[p * 2 + 1].cl = t[p * 2 + 1].cr = t[p * 2 + 1].add = t[p].add;
		t[p * 2].dat = t[p * 2 + 1].dat = 1;
		t[p].add = 0;
	}
}

void build(int p, int l, int r) {
	t[p].l = l;
	t[p].r = r;
	if (l == r) {
		t[p].dat = 1;
		t[p].cl = t[p].cr = a[l];
		return;
	}
	int mid = l + r >> 1;
	build(p * 2, l, mid);
	build(p * 2 + 1, mid + 1, r);
	t[p].cl = t[p * 2].cl;
	t[p].cr = t[p * 2 + 1].cr;
	t[p].dat = t[p * 2].dat + t[p * 2 + 1].dat;
	if (t[p * 2].cr == t[p * 2 + 1].cl) t[p].dat--;
}

int askColor(int p, int pos) {
	if (t[p].l == t[p].r) return t[p].cl;
	spread(p);
	int mid = t[p].l + t[p].r >> 1;
	if (pos <= mid) return askColor(p * 2, pos);
	else return askColor(p * 2 + 1, pos);
}

int askDat(int p, int l, int r) {
	if (l <= t[p].l && t[p].r <= r) return t[p].dat;
	spread(p);
	int mid = t[p].l + t[p].r >> 1, res = 0;
	if (l <= mid) res += askDat(p * 2, l, r);
	if (r > mid) res += askDat(p * 2 + 1, l, r);
	if (l <= mid && r > mid) res -= (t[p * 2].cr == t[p * 2 + 1].cl);
	return res;
}

void change(int p, int l, int r, int v) {
	if (l <= t[p].l && t[p].r <= r) {
		t[p].add = v;
		t[p].dat = 1;
		t[p].cl = t[p].cr = v;
		return;
	}
	spread(p);
	int mid = t[p].l + t[p].r >> 1;
	if (l <= mid) change(p * 2, l, r, v);
	if (r > mid) change(p * 2 + 1, l, r, v);
	t[p].cl = t[p * 2].cl;
	t[p].cr = t[p * 2 + 1].cr;
	t[p].dat = t[p * 2].dat + t[p * 2 + 1].dat;
	if (t[p * 2].cr == t[p * 2 + 1].cl) t[p].dat--;
}

void Rotate(int k) {
	if (!f) x += k;
	else x -= k;
}

void Swap(int x, int y) {
	int tx = askColor(1, x), ty = askColor(1, y);
	change(1, x, x, ty);
	change(1, y, y, tx);
}

void Paint(int l, int r, int x) {
	if (f) std::swap(l, r);
	if (l <= r) change(1, l, r, x);
	else change(1, l, n, x), change(1, 1, r, x);
}

int P(int p) {
	if (f) p = n - p + 2;
	return ((p - x) % n + n - 1) % n + 1;
}

int CountSegment(int l, int r) {
	int res = 0;
	if (f) std::swap(l, r);
	if (l > r) {
		res = askDat(1, l, n) + askDat(1, 1, r);
		if (t[1].cl == t[1].cr) res--;
	} else {
		res = askDat(1, l, r);
		if (l == 1 && r == n && t[1].cl == t[1].cr && res > 1) res--;
	}
	return res;
}

void Count() {
	int res = askDat(1, 1, n);
	if (t[1].cl == t[1].cr && res > 1) res--;
	printf("%d\n", res);
}

int main() {
	scanf("%d %d", &n, &c);
	for (int i = 1; i <= n; i++)
		scanf("%d", &a[i]);
	build(1, 1, n);
	scanf("%d", &q);
	for (; q; q--) {
		scanf("%s", op);
		int k, a, b, x;
		if (op[0] == 'R') {
			scanf("%d", &k);
			Rotate(k);
		} else if (op[0] == 'F') f ^= 1;
		else if (op[0] == 'S') {
			scanf("%d %d", &a, &b);
			a = P(a);
			b = P(b);
			Swap(a, b);
		}
		else if (op[0] == 'P') {
			scanf("%d %d %d", &a, &b, &x);
			a = P(a);
			b = P(b);
			Paint(a, b, x);
		} 
		else if (op[0] == 'C' && op[1] != 'S') Count();
		else {
			scanf("%d %d", &a, &b);
			a = P(a);
			b = P(b);
			printf("%d\n", CountSegment(a, b));
		}
	}
}