对两个火柴盒的火柴高度分别排序,当两火柴盒中排序序列号相等的火柴一一对应时距离最小
证明:
令a[1]<a[2] b[1]<b[2],
(a[1]-b[1])* (a[1]-b[1])+(a[2]-b[2])* (a[2]-b[2])-(a[1]-b[2])(a[1]-b[2])-(a[2]-b[1])(a[2]-b[1])
=-2a[1]*b[1]-2a[2]*b[2]+2a[1]*b[2]+2a[2]*b[1]
=2(a[1]-a[2])(b[2]-b[1])<0 得证。
找到第二列火柴中与第一列火柴啊a[i]的序列号相等的火柴b[j]的位置f[i],查找f[i]的逆序对的个数即为最后答案。
可用归并排序或者树状数组求逆序对
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=100005,mod=99999997;
int n;
struct node{
ll x;
int id;
}a[N],b[N];
int c[N],d[N],f[N],f1[N];
ll ans=0;
ll read(){
ll sum=0,f=1;
char ch=getchar();
while(ch>'9'||ch<'0'){
if(ch=='-')f=-1;
ch=getchar();
}
while(ch>='0'&&ch<='9'){
sum=(sum<<3)+(sum<<1)+ch-'0';
ch=getchar();
}
return sum*f;
}
int comp(node a,node b)
{
return a.x<b.x;
}
void merge(int l,int r)
{
if(l>=r) return;
int mid=(l+r)/2;
merge(l,mid);
merge(mid+1,r);
int i=l,j=mid+1,k=l-1;
while(i<=mid&&j<=r)
{
k++;
if(f[i]>f[j])
{
f1[k]=f[j];
ans+=mid-i+1;
ans%=mod;
j++;
}
else {
f1[k]=f[i];
i++;
}
}
while(i<=mid){k++;f1[k]=f[i];i++;}
while(j<=r){k++;f1[k]=f[j];j++;}
for(int i=l;i<=r;i++)
{
f[i]=f1[i];
}
}
int main(){
// freopen("match.in","r",stdin);
// freopen("match.out","w",stdout);
n=read();
for(int i=1;i<=n;i++)
{ a[i].x=read();
a[i].id=i;}
for(int i=1;i<=n;i++)
{ b[i].x=read();
b[i].id=i;}
sort(a+1,a+n+1,comp);
sort(b+1,b+n+1,comp);
for(int i=1;i<=n;i++)
{
c[a[i].id]=i;
d[i]=b[i].id;
}
for(int i=1;i<=n;i++){
f[i]=d[c[i]];
}
merge(1,n);
cout<<ans;
return 0;
}