P3939 数颜色

题目描述

小 C 的兔子不是雪白的，而是五彩缤纷的。每只兔子都有一种颜色，不同的兔子可能有 相同的颜色。小 C 把她标号从 1 到 nn 的 nn 只兔子排成长长的一排，来给他们喂胡萝卜吃。 排列完成后，第 ii 只兔子的颜色是 a_iai​。

俗话说得好，“萝卜青菜，各有所爱”。小 C 发现，不同颜色的兔子可能有对胡萝卜的 不同偏好。比如，银色的兔子最喜欢吃金色的胡萝卜，金色的兔子更喜欢吃胡萝卜叶子，而 绿色的兔子却喜欢吃酸一点的胡萝卜……为了满足兔子们的要求，小 C 十分苦恼。所以，为 了使得胡萝卜喂得更加准确，小 C 想知道在区间 [l_j,r_j][lj​,rj​] 里有多少只颜色为 c_jcj​ 的兔子。

不过，因为小 C 的兔子们都十分地活跃，它们不是很愿意待在一个固定的位置；与此同 时，小 C 也在根据她知道的信息来给兔子们调整位置。所以，有时编号为 x_jxj​ 和 x_j+1xj​+1 的两 只兔子会交换位置。 小 C 被这一系列麻烦事给难住了。你能帮帮她吗？

输入格式

从标准输入中读入数据。 输入第 1 行两个正整数 nn,mm。

输入第 2 行 nn 个正整数，第 ii 个数表示第 ii 只兔子的颜色 a_iai​。

输入接下来 mm 行，每行为以下两种中的一种：

• “1\ l_j\ r_j\ c_j1 lj​ rj​ cj​” ：询问在区间 [l_j,r_j][lj​,rj​] 里有多少只颜色为 c_jcj​ 的兔子；

• “2\ x_j2 xj​”： x_jxj​ 和 x_j+1xj​+1 两只兔子交换了位置。

输出格式

输出到标准输出中。

对于每个 1 操作，输出一行一个正整数，表示你对于这个询问的答案。

输入输出样例

输入 #1复制

6 5 
1 2 3 2 3 3  
1 1 3 2 
1 4 6 3  
2 3 
1 1 3 2  
1 4 6 3

输出 #1复制

1 
2 
2 
3 

思路：用可持久化线段树来维护颜色x的个数

当询问l~r的x的个数的时候我们就用版本r的x的个数减去版本l-1的x的个数

然后就是修改操作：

对于每次交换a[j]和a[j+1]这两个数，因为我们维护的是x的个数

那么个数改变的只有j版本的线段树：在a[j]处-1，在a[j+1]处+1

对于其他版本的线段树来说个数都不变，就不用修改其他版本的线段树了，交换一下a[j]和a[j+1]就行了

注意这题的数据范围，因为ci的值<=3e5，那么我们根节点的范围就是0~3e5+10

要算tr数组有多大，那么我们就根据他的内存250MB来开尽可能大的数：1024*1024*240/4/3

大约是2e7的样子

那么这道题就可以ac辽~

#include<bits/stdc++.h>
using namespace std;
const int N=3e5+10;
int n,m;
int a[N];
struct name{
	int l,r,sum;
}tr[20000000];
int root[N];
int idx;
//int build(int l,int r){
//	int p=++idx;
//	if(l==r)return p;
//	int mid=l+r>>1;
//	tr[p].l =build(l,mid);
//	tr[p].r =build(mid+1,r);
//	return p;
//}
int modify(int p,int l,int r,int x,int v){
	int q=++idx;
	tr[q]=tr[p];
	if(l==r){
		tr[q].sum +=v;
		return q;
	}
	int mid=l+r>>1;
	if(x<=mid) tr[q].l =modify(tr[p].l ,l,mid,x,v);
	else tr[q].r =modify(tr[p].r ,mid+1,r,x,v);
	tr[q].sum =tr[tr[q].l ].sum +tr[tr[q].r ].sum ;
	return q;
}
int query(int p,int q,int l,int r,int x){
	if(l==r) return tr[q].sum -tr[p].sum ;
	int mid=l+r>>1;
	if(x<=mid) return query(tr[p].l ,tr[q].l ,l,mid,x);
	else return query(tr[p].r ,tr[q].r ,mid+1,r,x);
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)scanf("%d",&a[i]);
//	root[0]=build(0,n+1);
	for(int i=1;i<=n;i++){
		root[i]=modify(root[i-1],0,N,a[i],1);
	}
//	for(int i=1;i<=n;i++){
//		printf("%d\n",query(root[n],1,n,a[i]));
//	}
//	cout<<endl;
	while(m--){
		int op,l,r,x;
		scanf("%d",&op);
		if(op==1){
			scanf("%d%d%d",&l,&r,&x);
			printf("%d\n",query(root[l-1],root[r],0,N,x));
		}else{
			int j;
			scanf("%d",&j);
			int t1=modify(root[j],0,N,a[j],-1);
			int t2=modify(t1,0,N,a[j+1],1);
			root[j]=t2;
			swap(a[j],a[j+1]);
		}
	}
	
	return 0;
}