505. 火柴排队

原题链接

思路

如下是画图分析的一些过程

在这里贪心的思路是排序，然后两个数组都是从小到大那样对应的话最终的答案可达到最小
而我们只能交换相邻的火柴，故在这里先假设一个简化版本，即A有序，而只需要对B进行操作，可发现，其实就是再在求逆序对数量
有了简化版本思路，我们再对原数组进行处理，这里想到将其转化为简化版，也就是将原数组映射为排好序的1,2,3,…这样的数组，然后相应的也映射到B数组，随后就可以进行操作了
要注意的一点，数据过大，故先将a[], b[]数组进行离散化

离散化的话一般是从下标1开始的，故题目中最好也用下标从1开始

代码如下

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 100010, MOD = 99999997;
int n, a[N], b[N], c[N], p[N];  //这里的p数组可理解为临时数组，用来当作容器为其他数组服务
//c[]数组为映射数组void work(int a[]) {for(int i = 1; i <= n; i++) p[i] = a[i];sort(p + 1, p + n + 1);for(int i = 1; i <= n; i++) a[i] = lower_bound(p + 1, p + n + 1, a[i]) - p;
}int merge_sort(int l, int r) {if(l >= r)  return 0;int mid = l + r >> 1;//这里的分治区间别忘了累加resint 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; i <= r; i++, j++) b[i] = p[j];return res;
}int main() {scanf("%d", &n);for(int i = 1; i <= n; i++) scanf("%d", &a[i]);for(int i = 1; i <= n; i++) scanf("%d", &b[i]);//首先，数据太大，进行数据的离散化（也是为了后期的映射对位）work(a), work(b);//然后，进行映射，将A数组映射为一个排好序的数组，B数组根据A数组数据也对应位置for(int i = 1; i <= n; i++) c[a[i]] = i;//对B的操作，在映射数组C[]中找b[i]的映射数据for(int i = 1; i <= n; i++) b[i] = c[b[i]];//最后，进行求最少交换次数（逆序对数量）int ans = merge_sort(1, n) % MOD;cout << ans;return 0;
}