题面
题目描述
涵涵有两盒火柴,每盒装有 n 根火柴,每根火柴都有一个高度。现在将每盒中的火柴各自排成一列,同一列火柴的高度互不相同,两列火柴之间的距离定义为: ,其中 ai表示第一列火柴中第 i 个火柴的高度,bi表示第二列火柴中第 i 个火柴的高度。
每列火柴中相邻两根火柴的位置都可以交换,请你通过交换使得两列火柴之间的距离最小。请问得到这个最小的距离,最少需要交换多少次?如果这个数字太大,请输出这个最小交换次数对 99,999,997 取模的结果。
输入
共三行,第一行包含一个整数 n,表示每盒中火柴的数目。
第二行有 n 个整数,每两个整数之间用一个空格隔开,表示第一列火柴的高度。
第三行有 n 个整数,每两个整数之间用一个空格隔开,表示第二列火柴的高度。
输出
输出共一行,包含一个整数,表示最少交换次数对 99,999,997 取模的结果。
样例输入
输入1:
4
2 3 1 4
3 2 1 4
输入2:
4
1 3 4 2
1 7 2 4
样例输出
输出1:
1
样例1说明:
最小距离是 0,最少需要交换 1 次,比如:交换第 1 列的前 2 根火柴或者交换第 2 列的前 2 根火柴。
输出2:
2
样例2说明:
最小距离是 10,最少需要交换 2 次,比如:交换第 1 列的中间 2 根火柴的位置,再交换第 2 列中后 2 根火柴的位置。
数据范围
对于 10%的数据, 1 ≤ n ≤ 10;
对于 30%的数据,1 ≤ n ≤ 100;
对于 60%的数据,1 ≤ n ≤ 1,000;
对于 100%的数据,1 ≤ n ≤ 100,000,0 ≤火柴高度≤ 2^31 − 1。
题解
这题其实很简单,我们知道若
a
1
<
a
2
,
b
1
<
b
2
a_1<a_2,b_1<b_2
a1<a2,b1<b2则显然
(
a
1
−
b
1
)
2
+
(
a
2
−
b
2
)
2
<
(
a
1
−
b
2
)
2
+
(
a
2
−
b
1
)
2
(a_1-b_1)^2+(a_2-b_2)^2<(a_1-b_2)^2+(a_2-b_1)^2
(a1−b1)2+(a2−b2)2<(a1−b2)2+(a2−b1)2
所以显然在a与b大小关系一一对应时
∑
i
=
1
n
(
a
i
−
b
i
)
2
\sum_{i=1}^n (a_i-b_i)^2
∑i=1n(ai−bi)2最小,
故将a、b排序
又因就近移动比单独的排序要好,所以可以建立如下关系
如样例
1 3 4 2
1 7 2 4
从小到大得
1
1
2
4
3
2
4
3
1_1\ 2_4\ 3_2\ 4_3
11 24 32 43
1
1
2
3
4
4
7
2
1_1\ 2_3\ 4_4\ 7_2
11 23 44 72
(大数字表示的是原数列中的数,小数字表示该数在原序列中的位置)
则可得C数组:
c
[
b
[
i
]
.
p
l
a
c
e
]
=
a
[
i
]
.
p
l
a
c
e
c[b[i].place]=a[i].place
c[b[i].place]=a[i].place
则答案就为C数组的逆序对个数。
归并排序或线段树即可。
代码
#include<bits/stdc++.h>
using namespace std;
#define mo 99999997
#define mid (l+r)/2
int i,j,n,m,k,o,p,c[100005];
long long ans;
struct node
{
int sz,wz;
}a[100005],b[100005];
inline int cmp(node x,node y)
{
return x.sz<y.sz;
}
inline void msort(int l,int r)
{
if (l==r) return ;
int d[100005];
msort(l,mid);
msort(mid+1,r);
int a2=l,b2=mid+1;
while (a2<=mid&&b2<=r)
{
if (c[a2]<c[b2])
{
d[a2+b2-mid-1]=c[a2++];
}
else
{
ans+=mid-a2+1;
d[a2+b2-mid-1]=c[b2++];
}
}
while (a2<=mid)
{
d[a2+b2-mid-1]=c[a2++];
}
while (b2<=r)
{
d[a2+b2-mid-1]=c[b2++];
}
for (int x=l;x<=r;x++) c[x]=d[x];
}
int main()
{
scanf("%d",&n);
for (i=1;i<=n;i++) {scanf("%d",&a[i].sz);a[i].wz=i;}
for (i=1;i<=n;i++) {scanf("%d",&b[i].sz);b[i].wz=i;}
sort(a+1,a+n+1,cmp);
sort(b+1,b+n+1,cmp);
for (i=1;i<=n;i++)
{
c[b[i].wz]=a[i].wz;
}
msort(1,n);
printf("%d",ans%mo);
}