P1966

[NOIP2013 提高组] 火柴排队

题目描述

涵涵有两盒火柴，每盒装有 $n$ 根火柴，每根火柴都有一个高度。现在将每盒中的火柴各自排成一列，同一列火柴的高度互不相同，两列火柴之间的距离定义为： $\sum (a_{i} - b_{i})^{2}$ 。

其中 $a_{i}$ 表示第一列火柴中第 $i$ 个火柴的高度， $b_{i}$ 表示第二列火柴中第 $i$ 个火柴的高度。

每列火柴中相邻两根火柴的位置都可以交换，请你通过交换使得两列火柴之间的距离最小。请问得到这个最小的距离，最少需要交换多少次？如果这个数字太大，请输出这个最小交换次数对 $10^{8} - 3$ 取模的结果。

对于 $100 %$ 的数据， $1 \leq n \leq 10^{5}$ ， $0 \leq$ 火柴高度 $< 2^{31}$ 。

分析

~~依题意模拟即可~~

注意到，可以随意对两个数组进行排序。

分析可知若需要 $\sum (a_{i} - b_{i})^{2}$ 最小，可以让两数组都有序（单调性相同）。

如果考虑对两个数组都进行若干次交换，情况太复杂了。

发现这完全可以转化为只对其中一个数组的交换（比如说 $b_{i}$ ）。

又注意到，这两个数组的对应位置是相对独立的，那么有以下做法：

对 $b_{i}$ 进行若干次交换，使得 $b_{i}$ 作为第 $k$ 大的数与 $a_{i}$ 同样也是第 $k$ 大的数相对应。
（第 $k$ 大指相对于各自的数组）

但是 $a_{i}$ 不是有序的，这是如果对 $b_{i}$ 进行排序，会很难处理，考虑进行转化。

假设 $a_{i}$ 一开始就是有序的，那么只需要对 $b_{i}$ 从小到大排序即可。

答案就是 $b_{i}$ 中逆序对的数量，用树状数组维护。

如何进行转化呢？需要重新指定数字间的大小关系。

关系是什么呢？从 $a_{i}$ 来。

如果认为 $a_{i}$ 是第 $i$ 小的数，就可以以此对 $b_{i}$ 进行排序了。

不要忘了进行离散化！（把值域缩到 $[1, n]$ ）

#include <bits/stdc++.h>
using namespace std;

const int N = 1e5 + 5;
const int mod = (114514+114514)*(-11+451-4)+114514+114*51*4+114*51+4+1/1-4+51*4;

int n, ans;
int a[N], b[N], to[N];

struct bitree{
    int c[N], res;
    inline int lb(int x) { return x & (-x); }
    inline void add(int x, int v){
        for (; x <= n; x += lb(x))
            c[x] += v;
    }
    inline int find(int x){
        for (res = 0; x; x -= lb(x))
            res += c[x];
        return res;
    }
    inline int find(int x, int y){
        return find(y) - find(x - 1);
    }
}tr;

map<int, int> mpa, mpb;

signed main(){
    
    cin >> n;
    for (int i = 1; i <= n; i++)
        cin >> a[i], mpa[a[i]] = i;
    
    for (int i = 1; i <= n; i++)
        cin >> b[i], mpb[b[i]] = i;
    
    sort(a + 1, a + 1 + n); sort(b + 1, b + 1 + n);
    
    for (int i = 1; i <= n; i++)
        to[mpa[a[i]]] = mpb[b[i]];
    
    for (int i = 1; i <= n; i++){
        ans = (ans + tr.find(n - to[i])) % mod;
        tr.add(n - to[i] + 1, 1);
    }
    
    cout << ans;
    
    
    return 0;
}