题目描述
涵涵有两盒火柴,每盒装有 n 根火柴,每根火柴都有一个高度。 现在将每盒中的火柴各自排成一列, 同一列火柴的高度互不相同, 两列火柴之间的距离定义为: ∑(ai-bi)^2
其中 ai 表示第一列火柴中第 i 个火柴的高度,bi 表示第二列火柴中第 i 个火柴的高度。
每列火柴中相邻两根火柴的位置都可以交换,请你通过交换使得两列火柴之间的距离最小。请问得到这个最小的距离,最少需要交换多少次?如果这个数字太大,请输出这个最小交换次数对 99,999,997 取模的结果。
【题目分析】
贪心的想法,同一位置的两个数在两个数组中排名相同,然后就要计算逆序对的个数了。树状数组又快又好写。
【代码】
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
struct node{
int pos;
int val;
}a[100001],b[100001];
int n;
int c[100001],tree[100001];
inline int getsum(int x)
{int k=x,ret=0; for (;k;k-=(k&(-k))) ret+=tree[k]; return ret;}
inline void add(int x)
{int k=x;for (;k<=n;k+=(k&(-k))) tree[k]++;}
inline bool cmp(node a,node b)
{return a.val<b.val;}
int main()
{
scanf("%d",&n);
for (int i=1;i<=n;++i) scanf("%d",&a[i].val),a[i].pos=i;
for (int i=1;i<=n;++i) scanf("%d",&b[i].val),b[i].pos=i;
sort(a+1,a+n+1,cmp);
sort(b+1,b+n+1,cmp);
for (int i=1;i<=n;++i) c[a[i].pos]=b[i].pos;
int ans=0;
for (int i=1;i<=n;++i)
{
add(c[i]);
ans+=i-getsum(c[i]);
ans%=99999997;
}
cout<<ans<<endl;
}