【JZOJ1214】【洛谷P4130】项链工厂【线段树】

题目大意:

题目链接:https://www.luogu.org/problemnew/show/P4130
一条项链包含 N 个珠子,每个珠子的颜色是 1,2,…,c 中的一种。项链

被固定在一个平板上,平板的某个位置被标记位置 1 ,按顺时针方向其他位置被记为 2,3,…,N。
在这里插入图片描述
你将要编写的软件系统应支持如下命令:

  1. R   k R\ k R k意为 R o t a t e   k Rotate\ k Rotate k。将项链在平板上顺时针旋转k 个位置, 即原来处于位置1 的珠子将转至位置k+1,处于位置2 的珠子将转至位置k+2,依次类推。
  2. F F F意为 F l i p Flip Flip。将平板沿着给定的对称轴翻转,原来处于位置1 的珠子不动,位置2 上的珠子与位置N 上的珠子互换,位置3 上的珠子与位置N-1 上的珠子互换,依次类推。
  3. S   i   j S\ i\ j S i j意为 S w a p   i , j Swap\ i, j Swap i,j。将位置i 上的珠子与位置j 上的珠子互换。
  4. P   I   j   x P\ I\ j\ x P I j x意为 P a i n t   i , j , x Paint\ i , j , x Paint i,j,x。将位置i 沿顺时针方向到位置j 的一段染为颜色x。
  5. C C C意为 C o u n t Count Count。查询当前的项链由多少个“部分”组成,我们称项链中颜色相同的一段为一个“部分”。
  6. C S   i   j CS\ i\ j CS i j意为 C o u n t S e g m e n t   i , j CountSegment\ i , j CountSegment i,j。查询从位置i沿顺时针方向到位置j 的一段中有多少个部分组成。

思路:

这就是那种算法简单思路简单细节贼多的题目 q w q qwq qwq
操作3456其实就是线段树的基本操作。维护每个区间左端点的值、右端点的值和这段区间的部分个数即可。
仔细观察发现,操作12都是没有改变这个数列的顺序的。仅仅是改变了位置。所以我们可以通过记录旋转次数以及是否翻转来从询问的点反推回初始的点。这样就可以直接在线段树上维护了。
细节较多,但是实现起来其实不算很难。注意细节即可。
代码不长,也就近200行而已。


代码:

#include <cstdio>
#include <string>
#include <cstring>
#include <iostream>
using namespace std;

const int N=500010;
int n,m,Q,sum,flag,a[N];
char ch[5];

struct Tree
{
	int l,r,lval,rval,cnt,lazy;
}tree[N*4];

int read()
{
	int d=0;
	char ch_=getchar();
	while (!isdigit(ch_)) ch_=getchar();
	while (isdigit(ch_))
		d=(d<<3)+(d<<1)+ch_-48,ch_=getchar();
	return d;
}

void write(int x)
{
	if (x>9) write(x/10);
	putchar(x%10+48);
}

void up(int x)  //利用子节点维护父节点
{
	tree[x].cnt=tree[x*2].cnt+tree[x*2+1].cnt;
	if (tree[x*2].rval==tree[x*2+1].lval) tree[x].cnt--;  //左右相同的连在一起的情况
	tree[x].lval=tree[x*2].lval; 
	tree[x].rval=tree[x*2+1].rval;
}

void pushdown(int x)
{
	if (tree[x].lazy)  //lazy标记下传
	{
		tree[x*2].lval=tree[x*2].rval=tree[x*2].lazy=tree[x].lazy;
		tree[x*2+1].lval=tree[x*2+1].rval=tree[x*2+1].lazy=tree[x].lazy;
		tree[x*2].cnt=tree[x*2+1].cnt=1;
		tree[x].lazy=0;
	}
}

void build(int x)  //建树不解释
{
	if (tree[x].l==tree[x].r)
	{
		tree[x].lval=tree[x].rval=a[tree[x].l];
		tree[x].cnt=1;
		return;
	}
	int mid=(tree[x].l+tree[x].r)/2;
	tree[x*2].l=tree[x].l;
	tree[x*2].r=mid;
	tree[x*2+1].l=mid+1;
	tree[x*2+1].r=tree[x].r;
	build(x*2); build(x*2+1);
	up(x);
}

int find(int x,int k)  //查询线段树中位置k的值
{
	if (tree[x].l==k && tree[x].r==k) return tree[x].lval;
	pushdown(x);
	int mid=(tree[x].l+tree[x].r)/2;
	if (k<=mid) return find(x*2,k);
		else return find(x*2+1,k);
}

void change(int x,int k,int val)  //单点修改
{
	if (tree[x].l==k && tree[x].r==k)
	{
		tree[x].lval=tree[x].rval=val;
		tree[x].cnt=1;
		return;
	}
	pushdown(x);
	int mid=(tree[x].l+tree[x].r)/2;
	if (k<=mid) change(x*2,k,val);
		else change(x*2+1,k,val);
	up(x);
}

void Paint(int x,int l,int r,int val)  //区间修改
{
	if (tree[x].l==l && tree[x].r==r)
	{
		tree[x].lval=tree[x].rval=tree[x].lazy=val;
		tree[x].cnt=1;
		return;
	}
	if (tree[x].l==tree[x].r) return;
	pushdown(x);
	int mid=(tree[x].l+tree[x].r)/2;
	if (r<=mid) Paint(x*2,l,r,val);
	else if (l>mid) Paint(x*2+1,l,r,val);
	else
	{
		Paint(x*2,l,mid,val);
		Paint(x*2+1,mid+1,r,val);
	}
	up(x);
}

void Swap(int x,int y)
{
	int t1=find(1,x),t2=find(1,y);
	change(1,y,t1); change(1,x,t2);  //将两者的值交换
}

int ask(int x,int l,int r)  //查询l~r中的段落个数
{
	if (tree[x].l==l && tree[x].r==r) return tree[x].cnt;
	pushdown(x);
	int mid=(tree[x].l+tree[x].r)/2;
	if (r<=mid) return ask(x*2,l,r);
	if (l>mid) return ask(x*2+1,l,r);
	int ans=ask(x*2,l,mid)+ask(x*2+1,mid+1,r);
	if (tree[x*2].rval==tree[x*2+1].lval) ans--;  //左右是相同的,连在一起是一个块
	return ans;
}

int get()  //转移回去
{
	int x;
	x=read();
	if (flag) x=n-x+2;  //有翻转
	return ((x-sum)%n+2*n-1)%n+1;
}

int main()
{
	n=read(); m=read();
	for (int i=1;i<=n;i++)
		a[i]=read();
	tree[1].l=1; tree[1].r=n;
	build(1);
	int x,y,k;
	Q=read();
	while (Q--)  //下面都是细节问题,注意就行了
	{
		scanf("%s",ch);
		if (ch[0]=='R')
		{
			k=read();
			if (flag) sum-=k;
				else sum+=k;
		}
		if (ch[0]=='F') flag^=1;
		if (ch[0]=='S')
		{
			x=get(); y=get();
			Swap(x,y);
		}
		if (ch[0]=='P')
		{
			x=get(); y=get(); k=read();
			if (flag) swap(x,y);
			if (x<=y) Paint(1,x,y,k);
			else
			{
				Paint(1,1,y,k);
				Paint(1,x,n,k);
			}
		}
		if (ch[0]=='C' && ch[1]=='S')
		{
			x=get(); y=get();
			if (flag) swap(x,y);
			if (x<=y) write(ask(1,x,y)),putchar(10);
			else 
			{
				int ans=ask(1,1,y)+ask(1,x,n);
				if (tree[1].lval==tree[1].rval && ans!=1) ans--;
				write(ans),putchar(10);
			}
		}
		if (ch[0]=='C' && ch[1]!='S')
		{
			int ans=ask(1,1,n);
			if (tree[1].lval==tree[1].rval && ans!=1) ans--;
			write(ans),putchar(10);
		}
	}
	return 0;
}
  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
洛谷P1168题目是关于中位数线段树解法的问题。中位数线段树解法可以通过维护两个堆来实现。一个是大根堆,一个是小根堆。每次插入元素时,根据一定的规则来维护这两个堆,使得大根堆的个数在一定情况下比小根堆多1或者相等。大根堆的最后一个元素即为中位数。具体的规则如下: 1. 如果大根堆和小根堆的个数相等,下一次插入的元素一定插入到大根堆。此时判断小根堆的堆顶是否大于当前元素x,如果是,则将小根堆的堆顶元素插入到大根堆,然后将x压入小根堆;否则直接将x压入大根堆。 2. 如果大根堆和小根堆的个数不相等,按照类似的规则进行操作。 通过以上规则,可以实现在每次插入元素时,维护两个堆的平衡,并且保证大根堆的最后一个元素即为中位数。 这种解法的时间复杂度为O(logN),其中N为序列的长度。 #### 引用[.reference_title] - *1* *2* [中位数(洛谷p1168)(堆/树状数组+二分/线段树+二分)](https://blog.csdn.net/qq_45604735/article/details/114382762)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^koosearch_v1,239^v3^insert_chatgpt"}} ] [.reference_item] - *3* [洛谷 P1168 中位数(权值线段树,离散化)](https://blog.csdn.net/qq_38232157/article/details/127594230)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^koosearch_v1,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值