Beyond the Void
BYVoid
NOI 2007 项链工厂

这个题要用线段树写,首先要满足的就是元素不能大规模移动。而题中的两种操作rotate和flip实际上是可以设一个偏移量来解决的。设flip为当前反转状态,offset为向右偏移量。对于R k命令,如果flip为false,使offset增加k,如果flip为true,使offset减少k。今后读入的位置i则应转化为绝对位置,如果flip为true,令i=N-i+2,然后再另i减去offset,超出范围后要增加或减少N的若干倍使1<=i<=N,得出i的绝对位置。

然后的问题就可以用一棵线段树来解决了,线段树上节点需要维护左边界颜色lc,右边界颜色rc,区间内颜色块数目cs,以及一个下传的标记same,表示该区间完全被染成了一个颜色。在区间查找或修改的时候,每经过一个节点要先标记下传,返回的时候还要自底向上计算lc,rc,cs。需要注意的是,自底向上计算的时候不要忘了,标记下传子节点。

由于整个项链是一个环,所以在执行C操作时,要判断最左边节点和最右边界点颜色是否相同,如果相同输出结果要为root.cs-1。另外,如果整个项链就是一个颜色,不能再减1了。

/* 
 * Problem: NOI2007 necklace
 * Author: Guo Jiabao
 * Time: 2009.6.5 16:30
 * State: Solved
 * Memo: 线段树 标记下传 上传计算
*/
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
using namespace std;
const int MAXN=500001;
struct SegmentTree
{
	struct Sgt_Node
	{
		Sgt_Node *left,*right;
		int a,b,lc,rc,cs,same;
	}SS[MAXN*2],*root;
	int SC;
	Sgt_Node * NewNode(int a,int b)
	{
		SS[++SC].same=0; SS[SC].a=a; SS[SC].b=b;
		SS[SC].lc=SS[SC].rc=0; SS[SC].cs=1;
		SS[SC].left = SS[SC].right = 0;
		return SS+SC;
	}
	void build(Sgt_Node *&x,int a,int b)
	{
		x=NewNode(a,b);
		if (a<b)
		{
			int m=(a+b) >> 1;
			build(x->left,a,m);
			build(x->right,m+1,b);
		}
	}
	void pushdown(Sgt_Node *x)
	{
		if (x->same)
		{
			x->lc = x->rc = x->same;
			x->cs = 1;
			if (x->a!=x->b)
				x->left->same = x->right->same = x->same;
			x->same=0;
		}
	}
	void update(Sgt_Node *x)
	{
		if (x->a!=x->b)
		{
			pushdown(x->left);
			pushdown(x->right);
			x->cs = x->left->cs + x->right->cs;
			if (x->left->rc == x->right->lc)
				x->cs --;
			x->lc = x->left->lc;
			x->rc = x->right->rc;
		}
	}
	int getcolor(Sgt_Node *x,int p)
	{
		pushdown(x);
		int m=(x->a + x->b) >> 1;
		if (x->a==x->b)
			return x->lc;
		else if (p<=m)
			return getcolor(x->left,p);
		else
			return getcolor(x->right,p);
	}
	void paint(Sgt_Node *x,int a,int b,int c)
	{
		int m=(x->a + x->b) >> 1;
		pushdown(x);
		if (x->a ==a && x->b==b)
		{
			x->same=c;
			pushdown(x);
		}
		else
		{
			if (b<=m)
				paint(x->left,a,b,c);
			else if (a>=m+1)
				paint(x->right,a,b,c);
			else
			{
				paint(x->left,a,m,c);
				paint(x->right,m+1,b,c);
			}
			update(x);
		}
	}
	int getcount(Sgt_Node *x,int a,int b)
	{
		int m=(x->a + x->b) >> 1,rs=0;
		pushdown(x);
		if (x->a ==a && x->b==b)
			return x->cs;
		else
		{
			if (b<=m)
				rs = getcount(x->left,a,b);
			else if (a>=m+1)
				rs = getcount(x->right,a,b);
			else
			{
				rs =  getcount(x->left,a,m);
				rs += getcount(x->right,m+1,b);
				if (x->left->rc == x->right->lc)
					rs--;
			}
		}
		update(x);
		return rs;
	}
}Segment;
int N,Q,C,offset;
bool flip;
inline void translate(int &a)
{
	if (flip)
		a=N-a+2;
	a-=offset;
	while (a<1) a+=N;
	while (a>N) a-=N;
}
int main()
{
	int i,k,c,a,b;
	char Cmd[5];
	freopen("necklace.in","r",stdin);
	freopen("necklace.out","w",stdout);
	scanf("%d%d",&N,&C);
	Segment.build(Segment.root,1,N);
	for (i=1;i<=N;i++)
	{
		scanf("%d",&c);
		Segment.paint(Segment.root,i,i,c);
	}
	scanf("%d",&Q);
	for (i=1;i<=Q;i++)
	{
		scanf("%s",Cmd);
		switch (Cmd[0])
		{
			case 'R':
				scanf("%d",&k);
				if (!flip)
					offset +=k ;
				else
					offset -=k ;
				while (offset<1) offset+=N;
				while (offset>N) offset-=N;
				break;
			case 'F':
				flip = !flip;
				break;
			case 'S':
				scanf("%d%d",&a,&b);translate(a);translate(b);
				c=Segment.getcolor(Segment.root,a);
				k=Segment.getcolor(Segment.root,b);
				Segment.paint(Segment.root,a,a,k);
				Segment.paint(Segment.root,b,b,c);
				break;
			case 'P':
				scanf("%d%d%d",&a,&b,&c);translate(a);translate(b);
				if (flip){k=a;a=b;b=k;}
				if (a<=b)
					Segment.paint(Segment.root,a,b,c);
				else
				{
					Segment.paint(Segment.root,a,N,c);
					Segment.paint(Segment.root,1,b,c);
				}
				break;
			case 'C':
				if (Cmd[1]=='S')
				{
					scanf("%d%d",&a,&b);
					translate(a);translate(b);
					if (flip){k=a;a=b;b=k;}
					if (a<=b)
						c=Segment.getcount(Segment.root,a,b);
					else
					{
						c=Segment.getcount(Segment.root,a,N);
						c+=Segment.getcount(Segment.root,1,b);
						if (Segment.root->lc == Segment.root->rc)
							c--;
					}
				}
				else
				{
					c=Segment.root->cs;
					if (c>1 && Segment.root->lc == Segment.root->rc)
						c--;
				}
				printf("%dn",c);
				break;
		}
	}
	return 0;
}
<h2><span class="mw-headline">项链工厂 </span></h2>

【问题描述】

T公司是一家专门生产彩色珠子项链的公司,其生产的项链设计新颖、款式多样、价格适中,广受青年人的喜爱。最近T公司打算推出一款项链自助生产系统,使用该系统顾客可以自行设计心目中的美丽项链。

该项链自助生产系统包括硬件系统与软件系统,软件系统与用户进行交互并控制硬件系统,硬件系统接受软件系统的命令生产指定的项链。该系统的硬件系统已经完成,而软件系统尚未开发,T公司的人找到了正在参加全国信息学竞赛的你,你能帮助T公司编写一个软件模拟系统吗?

一条项链包含N个珠子,每个珠子的颜色是1, 2, …, c中的一种。项链被固定在一个平板上,平板的某个位置被标记位置1,按顺时针方向其他位置被记为2,3,…,N。

<a class="image" title="Image:Necklace2007.jpg" href="http://www.ruvtex.cn/wiki/Image:Necklace2007.jpg"><img src="http://www.ruvtex.cn/mw/images/7/77/Necklace2007.jpg" alt="Image:Necklace2007.jpg" width="429" height="447" /></a>

你将要编写的软件系统应支持如下命令:

<table border="1">
<tbody>
<tr>
<td width="155">命令</td>
<td width="144">参数限制</td>
<td width="326">内容</td>
</tr>
<tr>
<td>R k</td>
<td>0&lt;k&lt;N</td>
<td>意为Rotate k。将项链在平板上顺时针旋转k个位置, 即原来处于位置1的珠子将转至位置k+1,处于位置2的珠子将转至位置k+2,依次类推。</td>
</tr>
<tr>
<td>F</td>
<td></td>
<td>意为Flip。将平板沿着给定的对称轴翻转,原来处于位置1的珠子不动,位置2上的珠子与位置N上的珠子互换,位置3上的珠子与位置N-1上的珠子互换,依次类推。</td>
</tr>
<tr>
<td>S i j</td>
<td>1≤i , j≤N</td>
<td>意为Swap i , j。将位置i上的珠子与位置j上的珠子互换。</td>
</tr>
<tr>
<td>P i j x</td>
<td>1≤i , j≤N, x≤c</td>
<td>意为Paint i , j , x。将位置i沿顺时针方向到位置j的一段染为颜色x。</td>
</tr>
<tr>
<td>C</td>
<td></td>
<td>意为Count。查询当前的项链由多少个“部分”组成,我们称项链中颜色相同的一段为一个“部分”</td>
</tr>
<tr>
<td>CS i j</td>
<td>1≤i , j≤N</td>
<td>意为CountSegment i , j。查询从位置i沿顺时针方向到位置j的一段中有多少个部分组成。</td>
</tr>
</tbody></table>

【输入文件】

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

【输出文件】

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

【输入样例】
<pre>5 3
1 2 3 2 1
4
C
R 2
P 5 5 2
CS 4 1</pre>
【输出样例】
<pre>4
1</pre>
【评分方法】

本题没有部分分,你的程序的输出只有和标准答案完全一致才能获得满分, 否则不得分。

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

上次修改时间 2017-05-22

相关日志