洛谷 P1966 [NOIP 2013 提高组] 火柴排队-普及+/提高-CSDN博客

本文链接：https://blog.csdn.net/weixin_37253733/article/details/148892329

题目背景

NOIP2013 提高组 D1T2

题目描述

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

其中 $a_i$ 表示第一列火柴中第 $i$ 个火柴的高度， $b_i$ 表示第二列火柴中第 $i$ 个火柴的高度。

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

输入格式

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

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

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

输出格式

一个整数，表示最少交换次数对 $10^8-3$ 取模的结果。

输入输出样例 #1

输入 #1

4
2 3 1 4
3 2 1 4

输出 #1

输入输出样例 #2

输入 #2

4
1 3 4 2
1 7 2 4

输出 #2

说明/提示

输入输出样例说明一

最小距离是 $ 0$，最少需要交换 $1$ 次，比如：交换第 $1 $ 列的前 $ 2$ 根火柴或者交换第 $2$ 列的前 $2 $ 根火柴。

输入输出样例说明二

最小距离是 $10$ ，最少需要交换 $2$ 次，比如：交换第 $1$ 列的中间 $2$ 根火柴的位置，再交换第 $2$ 列中后 $2$ 根火柴的位置。

数据范围

对于 $10%10\%$ 的数据， $\leq n \leq 10$ ；

对于 $30%30\%$ 的数据， $\leq n \leq 100$ ；

对于 $60%60\%$ 的数据， $\leq n \leq 10^3$ ；

对于 $100%100\%$ 的数据， $\leq n \leq 10^5$ ， $\leq a_i,b_i < 2^{31}$ 且对于任意 $1≤i<j≤n1\le i<j\le n$ ， $ai≠aja_i\neq a_j$ ， $bi≠bjb_i\neq b_j$ 。

solution

其实是求逆序数，只不过比较的标准是另一列的次序，即将第二列排成第一列一样的次序

先求出第一列和第二列的次序。 id[i] : a中第 i 名的位置 idx[i]：b中第 i 名的位置
f[id[i]] = idx[i]: a 中第 i 名位置目标位置是 b 中第 i 名的位置
求 f 的逆序数

代码

#include <iostream>
#include "bit"
#include "vector"
#include "unordered_set"
#include "set"
#include "queue"
#include "algorithm"
#include "bitset"
#include "cstring"

using namespace std;

/*
 * 其实是求逆序数，只不过比较的标准是另一列的次序，即将第二列排成第一列一样的次序
 * 1 先求出第一列和第二列的次序。 id[i] : a中第 i 名的位置  idx[i]：b中第 i 名的位置
 * 2 f[id[i]] = idx[i]:  a 中第 i 名位置目标位置为 b 中第 i 名的位置
 * 3 求 f 的逆序数
 */


const int N = 1e5 + 1, M = 1e8 - 3;
int n, a[N], b[N], id[N], idx[N], iid[N], f[N], t[N];
long long ans;

void merge(int low, int mid, int high) {
    int i = low, j = mid + 1, k = low;
    while (i <= mid && j <= high) {
        if (f[i] < f[j]) {
            t[k++] = f[i++];
        } else {
            t[k++] = f[j++];
            ans += (mid - i + 1);
        }
    }
    while (i <= mid) t[k++] = f[i++];
    while (j <= high) t[k++] = f[j++];

    memcpy(f + low, t + low, (high - low + 1) * 4);
}

void merge_sort(int low, int high) {
    if (low < high) {
        int mid = (low + high) >> 1;
        merge_sort(low, mid);
        merge_sort(mid + 1, high);
        merge(low, mid, high);
    }
}

int main() {
    cin >> n;

    for (int i = 0; i < n; i++) cin >> a[i], id[i] = i;
    sort(id, id + n, [&](int i, int j) {
        return a[i] < a[j];
    });

    for (int i = 0; i < n; i++) cin >> b[i], idx[i] = i;
    sort(idx, idx + n, [&](int i, int j) {
        return b[i] < b[j];
    });

    for (int i = 0; i < n; i++) f[id[i]] = idx[i];

    merge_sort(0, n - 1);
    cout << ans % M;

}