样例
10
A ;10 1 5 2 7 4 9 3 6 8
B:7 5 1 8 10 4 6 2 3 9
两队火柴大小关系应该对应,如果按照A的顺序,应得到B的位置关系为 5 3 2 10 1 6 9 4 8 7.
用save[]存下来 求次顺序用3次sort()即可。
再用归并排序求逆序对。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
#define maxn 110000
int n,save[maxn],L[maxn],R[maxn];
long long ans;
struct node
{
int num,loc;
}a[maxn],b[maxn];
struct node2
{
int a,b;
}c[maxn];
bool cmp(node aa,node bb)
{
return aa.num<bb.num;
}
bool cmp2(node2 aa,node2 bb)
{
return aa.a<bb.a;
}
void init()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i].num);
a[i].loc=i;
}
for(int i=1;i<=n;i++)
{
scanf("%d",&b[i].num);
b[i].loc=i;
}
sort(a+1,a+1+n,cmp);
sort(b+1,b+1+n,cmp);
for(int i=1;i<=n;i++)
{
c[i].a=a[i].loc;
c[i].b=b[i].loc;
}
sort(c+1,c+1+n,cmp2);
for(int i=1;i<=n;i++)
{
save[i]=c[i].b;
}
}
void merge(int l,int mid,int r)
{
int n1 = mid-l+1,n2 = r-mid;
int i,j,k;
for(i = 1;i <= n1; i++)
L[i] = save[l+i-1];
for(j = 1;j <= n2; j++)
R[j] = save[mid+j];
L[n1+1] = 0x3f3f3f3f;
R[n2+1] = 0x3f3f3f3f;
i = 1;j = 1;
for(k = l;k <= r;k ++)
{
if(L[i]<=R[j])
save[k] = L[i++];
else
{
save[k] = R[j++];
ans+=(n1 -i + 1);
}
}
}
void merge_sort(int l,int r)
{
if(l<r)
{
int mid = (l+r)/2;
merge_sort(l,mid);
merge_sort(mid+1,r);
merge(l,mid,r);
}
}
int main()
{
init();
merge_sort(1,n);
ans=ans%99999997;
printf("%lld",ans);
return 0;
}