[CF1093E]Intersection of Permutations

[CF1093E]Intersection of Permutations

题目大意：

给定两个长度为\(n(n\le2\times10^5)\)的排列\(A,B\)。\(m(m\le2\times10^5)\)次操作，操作分为以下两种：

1. 询问有多少同时在\(A_{[x,y]}\)和\(B_{[l,r]}\)中出现的数。
2. 交换\(B_x\)与\(B_y\)。

思路：

令\(v[a[i]]=i,b[i]=v[b[i]]\)，这样询问就变成\([x,y]\)中有多少数在\(B_{[l,r]}\)中出现。

用树状数组+平衡树维护每个区间出现哪些数，询问时在平衡树中查找排名即可。

源代码：

#include<cstdio>
#include<cctype>
#include<climits>
#include<ext/pb_ds/tree_policy.hpp>
#include<ext/pb_ds/assoc_container.hpp>
inline int getint() {
	register char ch;
	while(!isdigit(ch=getchar()));
	register int x=ch^'0';
	while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
	return x;
}
const int N=2e5+1;
int n,v[N],a[N];
using namespace __gnu_pbds;
using RBTree=tree<int,null_type,std::less<int>,rb_tree_tag,tree_order_statistics_node_update>;
class FenwickTree {
	private:
		RBTree t[N];
		int lowbit(const int &x) const {
			return x&-x;
		}
		int query(int p,const int &x,const int &y) const {
			int ret=0;
			for(;p;p-=lowbit(p)) {
				const int L=*t[p].lower_bound(x);
				const int R=*std::prev(t[p].upper_bound(y));
				if(L<=R) ret+=t[p].order_of_key(R)-t[p].order_of_key(L)+1;
			}
			return ret;
		}
	public:
		void init() {
			for(register int i=1;i<=n;i++) {
				t[i].insert(INT_MIN);
				t[i].insert(INT_MAX);
			}
		}
		void insert(int p,const int &x) {
			for(;p<=n;p+=lowbit(p)) {
				t[p].insert(x);
			}
		}
		void erase(int p,const int &x) {
			for(;p<=n;p+=lowbit(p)) {
				t[p].erase(x);
			}
		}
		int query(const int &l,const int &r,const int &x,const int &y) const {
			return query(r,x,y)-query(l-1,x,y);
		}
};
FenwickTree t;
int main() {
	n=getint();
	const int m=getint();
	for(register int i=1;i<=n;i++) {
		v[getint()]=i;
	}
	for(register int i=1;i<=n;i++) {
		a[i]=v[getint()];
	}
	t.init();
	for(register int i=1;i<=n;i++) {
		t.insert(i,a[i]);
	}
	for(register int i=0;i<m;i++) {
		const int opt=getint(),x=getint(),y=getint();
		if(opt==1) {
			const int l=getint(),r=getint();
			printf("%d\n",t.query(l,r,x,y));
		}
		if(opt==2) {
			t.erase(x,a[x]);
			t.erase(y,a[y]);
			std::swap(a[x],a[y]);
			t.insert(x,a[x]);
			t.insert(y,a[y]);
		}
	}
	return 0;
}