题目描述
涵涵有两盒火柴,每盒装有 n 根火柴,每根火柴都有一个高度。 现在将每盒中的火柴各自排成一列, 同一列火柴的高度互不相同, 两列火柴之间的距离定义为:∑(ai-bi)2
其中ai 表示第一列火柴中第i个火柴的高度,bi表示第二列火柴中第 i 个火柴的高度。
每列火柴中相邻两根火柴的位置都可以交换,请你通过交换使得两列火柴之间的距离最小。请问得到这个最小的距离,最少需要交换多少次?如果这个数字太大,请输出这个最小交换次数对99,999,997取模的结果。
输入格式
共三行,第一行包含一个整数n,表示每盒中火柴的数目。
第二行有n个整数,每两个整数之间用一个空格隔开,表示第一列火柴的高度。
第三行有n个整数,每两个整数之间用一个空格隔开,表示第二列火柴的高度。
输出格式
一个整数,表示最少交换次数对 99,999,997 取模的结果。
输入输出样例
4
2 3 1 4
3 2 1 4
4
1 3 4 2
1 7 2 4
说明/提示
【输入输出样例说明1】
最小距离是0,最少需要交换 1 次,比如:交换第 1列的前2 根火柴或者交换第 2 列的前 2根火柴。
【输入输出样例说明2】
最小距离是 10,最少需要交换2次,比如:交换第1 列的中间22根火柴的位置,再交换第2 列中后 2 根火柴的位置。
【数据范围】
对于 10\%的数据, 1 ≤ n ≤ 10;
对于 30\%的数据,1 ≤ n ≤ 100;
对于 60\%的数据,1 ≤ n ≤ 1,000;
对于 100\%的数据,1 ≤ n ≤ 100,000,0 ≤火柴高度≤ maxlongint
引导:
首先看到这道题我们,我们先好好动下脑经想:如何排序火柴才能将∑(a_i−b_i)^2 最大值求出来。
思路 :
所以我们应该容易想到将两个队列先进行排序,比如说将两个队列的最大的元素对齐,次大的元素对齐,以此类推,就可以发现 | a - b | 的值就会是最小的,那么这样 ∑(ai-bi)2的值是最小的。
这里我们提供两种代码:线段树 和 树状数组 (其实是同一种思想)
如何实现代码:
-
使用是树状数组时前面的模板是通用的,然后主函数理只需要将两个数组排序,再进行前面说到的匹配,就好了。
-
使用线段树时,也是一样先模板敲好,接下来最最关键的就是离散化, 观察题目,我们可以发现,所谓的求最少次数交换位置,因为离散化好后,不就是求这个数组中的逆序对?于是,代码就明确了!
何谓离散化?离散化就是不在乎这个数组内具体某一个值的大小(在这里我们称之为绝对数值),而是在乎这个数组内的相对大小(相对数值,也可以理解为排序后的排名)。比如下面有两个数组
A :100 200 399 488
B :938 132 144 77
在这里我们可以看到A【1】 = 488,A【2】 = 399,以此类推,说明A数组的离散化后 488** 所对应的值就是 1。 B 数组同理。
那么我们就可以很轻松的敲出代码即可。
若还是有不懂离散化的,下面代码里有详解。
(个人意见:跟着代码走更好理解。)
二维偏序做法:
我们都知道将两个序列按增序排列时可得到最小距离,但这样可能有些多余的交换(比如两个序列同时将两个相同位置的数作了相同的交换,交换到相同位置)
做法:先将一个序列变成增序,另一个序列跟着做同样的交换(可用结构体),得到一个序列,则将该序列变为增序的操作次数(也就是求该序列的逆序对数)就是将两个序列变为增序两两对应的操作次数
树状数组:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#define mod 99999997
using namespace std;
typedef long long LL;
int n, ans;
int c[1000009], d[1000009];
int a[1000009], b[1000009];
int q[1000009];
bool cmp1(int i, int j) {return a[i] < a[j];}
bool cmp2(int i, int j) {return b[i] < b[j];}
namespace Tree{
int low[1000009];
LL lowbit(LL x) {return -x&x;}
LL sum(LL x){
LL ret = 0;
while (x > 0){
ret += low[x];
x -= lowbit(x);
}
return ret;
}
void add(LL x, LL d){
while (x <= n){
low[x] += d;
x += lowbit(x);
}
}
}
using namespace Tree;
void solve()
{
ans = 0;
for (int i = n; i >= 1; i--){
ans += sum(q[i] - 1);
add(q[i], 1);
if (ans >= mod) ans -= mod;
}
printf("%d", (ans % mod + mod) % mod);
}
int main()
{
scanf("%d", &n);
for (int i = 1; i <= n; i++) scanf("%d", &a[i]), c[i] = i;
for (int i = 1; i <= n; i++) scanf("%d", &b[i]), d[i] = i;
sort(c + 1, c + n + 1, cmp1);
sort(d + 1, d + n + 1, cmp2);
memset(q, 0, sizeof(q));
for (int i = 1; i <= n; i++){
q[c[i]] = d[i];
}
solve();
return 0;
}
线段树:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cmath>
#include<set>
#include<cstring>
using namespace std;
//head
//时间复杂度: O(nlogn)
//这个题目又是求逆序对的。。。
#define lsn l,mid,rt<<1
#define rsn mid+1,r,rt<<1|1
const int N = 1000005;
const int mod = 99999997;
int n;
int v[N], a[N], b[N], c[N], pos[N], tree[N<<2];
void add(int l, int r, int rt, int p, int v){
if(l == r){
tree[rt] += v;
return;
}
int mid = (l + r) >> 1;
if(p <= mid) add(lsn, p, v);
else add(rsn, p, v);
tree[rt] = tree[rt<<1|1] + tree[rt<<1];
}
int query(int l, int r, int rt, int p){
if(r <= p) return tree[rt];
if(l > p) return 0;
int ans = 0;
int mid = (l + r) >> 1;
ans += query(lsn, p);
ans += query(rsn, p);
return ans;
}
int main(){
scanf("%d", &n);
for(int i = 1; i <= n; ++i){
scanf("%d", a+i);
v[i] = a[i];
}
sort(v+1, v+n+1);
for(int i = 1; i <= n; ++i){
a[i] = lower_bound(v+1, v+n+1, a[i]) - v; //离散化,就是我们只需要a[i]中每个元素的相对排名,并不关心它们的具体的数值大小
pos[a[i]] = i; //记录排名第a[i]的位置
}
for(int i = 1; i <= n; ++i){
scanf("%d", b+i);
v[i] = b[i];
}
sort(v+1, v+n+1);
for(int i = 1; i <= n; ++i) b[i] = lower_bound(v+1, v+n+1, b[i]) - v; //同样对b数组进行离散化
for(int i = 1; i <= n; ++i){
c[i] = pos[b[i]]; //用c数组来表示b数组中的第i个元素 应该放置的位置
}
//求逆序对
long long ans = 0;
for(int i = 1; i <= n; ++i){
int num = query(1, n, 1, c[i]);
ans = (ans + i - 1 - num) % mod;
add(1, n, 1, c[i], 1);
}
printf("%lld
", ans);
return 0;
}