题目描述 涵涵有两盒火柴,每盒装有 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; }