[NOIP2013 提高组] 火柴排队

最新推荐文章于 2025-11-23 17:00:00 发布

原创最新推荐文章于 2025-11-23 17:00:00 发布 · 443 阅读

13 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#算法

本文介绍了一种算法方法，通过归并排序计算两列高度不同的火柴的逆序对，以确定最少交换次数，使两列火柴间的距离最小。最后结果对特定大数取模。

涵涵有两盒火柴，每盒装有 n 根火柴，每根火柴都有一个高度。

现在将每盒中的火柴各自排成一列，同一列火柴的高度互不相同，两列火柴之间的距离定义为：

其中 ai 表示第一列火柴中第 i 个火柴的高度，bi 表示第二列火柴中第 i 个火柴的高度。

每列火柴中相邻两根火柴的位置都可以交换，请你通过交换使得两列火柴之间的距离最小。

请问得到这个最小的距离，最少需要交换多少次？

如果这个数字太大，请输出这个最小交换次数对 99,999,99799,999,997 取模的结果。

输入格式

共三行，第一行包含一个整数 n，表示每盒中火柴的数目。

第二行有 n 个整数，每两个整数之间用一个空格隔开，表示第一列火柴的高度。

第三行有 n 个整数，每两个整数之间用一个空格隔开，表示第二列火柴的高度。

输出格式

输出共一行，包含一个整数，表示最少交换次数对 99,999,99799,999,997 取模的结果。

数据范围

1≤n≤105,
0≤火柴高度≤231−10≤火柴高度≤231−1

输入样例：

4
2 3 1 4
3 2 1 4

输出样例：

首先分析题目，肯定是要往逆序对这个方向上去靠，求逆序对可以用归并排序或者树状数组。

我们可以发现本题对于火柴的移动，可以简化成：完全固定一边，只移动一边。接着我们只要计算，将另外一遍移动到对应位置需要几次。再通过离散化操作和哈希索引，将数组转换成用归并排序求逆序对的模板，即可完成本题。

给定两个数组 a 和 b，分别代表两列火柴的高度。

a: 2 3 1 4 b: 3 2 1 4
首先，我们可以通过数组 a 得到数组 p。数组 p 的作用是根据数组 a 的大小，构造一个新的数组，使得数组 p 中的值越小，对应的数组 a 中的值越小。

p: 1 2 0 3
然后，我们继续对数组 b 进行同样的处理，得到新的数组 c。数组 c 的作用是构造一个映射关系，使得数组 a 变成有序的，并且通过这个关系来对数组 b 确定顺序。

c: 2 0 1 3
构造完数组 c 后，我们可以通过数组 c 来对数组 b 进行映射，从而得到有序的数组 b。

b: 0 2 1 3
接下来，我们可以对数组 b 进行归并排序，统计其中的逆序对。逆序对的数量表示需要交换的次数，从而得到最小的交换次数。

#include <iostream>
#include <algorithm>
#include <unordered_map>

using namespace std;

typedef long long ll;

const int N = 1e5 + 10, mod = 99999997;

int n;
int top[N], bom[N], tem[N];
unordered_map<int, int>th, bh;
int en[N];
ll res;

void merge_sort(int l, int r) {
    if (l >= r) return ;
    int mid = l + r >> 1;
    merge_sort(l, mid), merge_sort(mid + 1, r);
    int j = l, i = mid + 1, k = 0;
    while (j <= mid && i <= r) {
        if (en[j] <= en[i]) tem[k ++ ] = en[j ++ ];
        else {
            tem[k ++] = en[i ++ ];
            res = (res + mid - j + 1) % mod;
        }
    }
    while (j <= mid) tem[k ++ ] = en[j ++ ];
    while (i <= r) tem[k ++ ] = en[i ++ ];
    for (int i = 0, j = l; j <= r; j ++ , i ++ ) en[j] = tem[i];
}


void output(int a[]) {
    for (int i = 0; i < n; i ++ ) cout << a[i] << ' ';
    cout << endl;
}

int main() {
    cin >> n;
    for (int i = 0; i < n; i ++ ) {
        cin >> top[i];
        th[top[i]] = i;
    }
    
    for (int i = 0; i < n; i ++ ) {
        cin >> bom[i];
        bh[bom[i]] = i;
    }
    sort(top, top + n);
    sort(bom, bom + n);
    int gjb[N], gjt[N];
    for (int i = 0; i < n; i ++ ) gjb[i] = bh[bom[i]];
    for (int i = 0; i < n; i ++ ) gjt[i] = th[top[i]];
    unordered_map<int, int>hiem;
    for (int i = 0; i < n; i ++ ) hiem[gjb[i]] = gjt[i];

    sort(gjb, gjb + n);
    for (int i = 0; i < n; i ++ ) en[i] = hiem[gjb[i]];
    merge_sort(0, n - 1);
    cout << res;
}