4.2练习 火柴排队 题解

这是一道难度 至少对于我来说 非常之大的一道题目，在听懂之后，想记录一下 并重新通过写博客的方式 过一遍思路。

题目如下：

问题的解释大概如下：有两盒火柴，每盒有n根，每个火柴的高度不同，现在你要做的是尽可能的让两列火柴的距离最小，定义两列火柴的距离为：1~n所有火柴的差的平方和。

这道题目我一开始独立思考的时候，大概想到了下面的东西：

首先题目要求我们算的是最少多少次移动之后，两个序列的对应位置的差值的平方之和最小，那么我们比较容易能够想到：当a，b两个数组通过相同的升降序规则排列的情况下，就是对应着最小平方和，那么我们继续往下面想，但是发现我们如果把两个数组分别计算排序总共需要移动多少次的话，似乎时间复杂度就过高了，所以我一直在想一种转化的方式，可惜还是没想出来。

下面是这道题目的正确思路做法之一：有一些抽象也有一些绕脑子，希望读者能够仔细体会：

由于题目要求我们算的是最少的交换次数，而不是要我们输出最小差的平方和的值，因此我们容易思考到，a和b数组对应的数字之间，其实只需要考虑相对关系，也就是说，a中第几小的数字对应b中第几小的数字即可，而数组中元素具体的值其实并不重要，那么思路到这里，我们就可以考虑使用离散化的方式来进行解答了，我们不妨将a，b数组里面的所有元素都转换成从1到n的数字形式，离散化之后其值就代表其是第几小也就是说假设我a数组一开始最小的数字是10那么离散化之后它的值就是1，代表是a数组中原本最小的数字，这样转化之后他们同样等价于原问题，然后将a数组通过映射的方式排好序并利用一个中间数组c，来充当a，b数组的中间桥梁，将b中元素以与a数组同样的映射方式，这样映射之后，a数组就转化成了一个有序的数组1到n，而我们需要求的答案也就是将b数组排成由小到大的序列一共需要交换多少次，也就是求逆序对。

代码如下：

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+100,MOD =99999997;
int n;
int a[N],b[N],c[N],p[N];
void work(int a[]){
	for(int i = 1;i <= n;i++) p[i] = i;
	sort(p+1,p+n+1,[&](int x,int y){
		return a[x] < a[y];
	});
	for(int i=1;i<=n;i++) a[p[i]] = i;
}
int merge_sort(int l,int r){
	if(l>=r) return 0;
	int mid = l + r >> 1;
    int res = (merge_sort(l,mid) + merge_sort(mid+1,r)) % MOD;
    int i=l,j=mid+1,k=0;
    while(i <= mid && j <= r){
    	if(b[i] < b[j]) p[k++] = b[i++];
    	else p[k++]=b[j++],res=(res + mid - i + 1) % MOD;
	}
	while(i<=mid) p[k++] = b[i++];
	while(j<=r) p[k++] = b[j++];
	for(int i=l,j=0;j<k;i++,j++) b[i] = p[j];
    return res;
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	cout.tie(nullptr);
	cin>>n;
	for(int i = 1;i <= n;i++)
	cin>>a[i];
	for(int i = 1;i <= n;i++)
	cin>>b[i];
	work(a),work(b);
	for(int i = 1;i <= n;i++)
	c[a[i]] = i;
	for(int i = 1;i <= n;i++)
	b[i] = c[b[i]]; 
    cout<<merge_sort(1,n)<<endl;
    return 0;
}

感谢观看，希望对读者有所帮助。