P1966 火柴排队

题面:https://www.luogu.org/problem/P1966

本题就是要将a序列第k大的元素和序列b中第k大的元素的位置换成一样的.
那么可以将a,b数组分别离散化,此时只要将a,b数组换成相同的即可.
令q[a[i]]=b[i],若a数组和b数组完全相同,则q[a[i]]=a[i],即q[i]=i.
那么就是要将q数组调成上升序列,那么求逆序对个数即为解.
注意:此处为归并排序求逆序对.

Code:
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=100002,M=99999997;
struct mt{
	int x,num;
}a[N],b[N];
int n,i,j,k,d[N],c[N],s;
bool cmp(mt p,mt q){
	if(p.x!=q.x)
		return p.x<q.x;
	return p.num<q.num;
}
void gb(int l,int r){
	if(l==r)
		return;
	int mid=l+r>>1;
	gb(l,mid);
	gb(mid+1,r);
	i=l;j=mid+1;k=0;
	while(i<=mid&&j<=r){
		if(c[i]<c[j])
			d[++k]=c[i++],s=(s+j-mid-1)%M;
		else
			d[++k]=c[j++];
	}
	while(i<=mid)
		d[++k]=c[i++],s=(s+j-mid-1)%M;
	while(j<=r)
		d[++k]=c[j++];
	for(i=1,j=l;i<=k;i++,j++)
		c[j]=d[i];
	return;
}
int main(){
	scanf("%d",&n);
	for(i=0;i<n;i++)
		scanf("%d",&a[i].x),a[i].num=i;
	for(i=0;i<n;i++)
		scanf("%d",&b[i].x),b[i].num=i;
	sort(a,a+n,cmp);	
	sort(b,b+n,cmp);
	for(i=0;i<n;i++)
		c[b[i].num]=a[i].num;
	gb(0,n-1);
	printf("%d",s);
	return 0;
}