每日一题 第六十二期 洛谷 火柴排列

[NOIP2013 提高组] 火柴排队

题目背景

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

1

样例 #2

样例输入 #2

4
1 3 4 2
1 7 2 4

样例输出 #2

2

提示

输入输出样例说明一

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

输入输出样例说明二

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

数据范围

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

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

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

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

AC代码：

#include<iostream>
#include<algorithm>
using namespace std;

const int mod = 99999997;
const int N = 1e5 + 10;
int n;
int a[N], b[N], c[N];
int tem[N], p[N];
void work(int a[])//新的离散化的方式
{
    for(int i = 1; i <= n; i ++) p[i] = i;//相当于初始化
    sort(p + 1, p + 1 + n, [&](int x, int y)
    {
        return a[x] < a[y]; 
    });//p的排序是以a的相对关系来进行排序的，大小关系即p显示的为a元素之间相对的大小
    for(int i = 1; i <= n; i ++) a[p[i]] = i;//在将他们的映射关系转换到对应的位置//相当于第几大的数对应第几个位置
    //此时a[i]代表第i个大的数在数组中的位置是多少
}//进行离散化

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

int main()
{
    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;//相当于将a[i]的映射关系弄到c中;
    for(int i = 1; i <= n; i ++) b[i] = c[b[i]];//相当于根据a中的关系来确定b中的关系
    cout << mer(b, 1, n) << endl;;
    return 0;
}