AcWing 505 火柴排队

AcWing 505

解题收获

给每个数找到它该去的位置，然后排序即可，注意题目中数据元素不会重复。

题目描述

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

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

∑i=1n(ai−bi)2
，其中 ai 表示第一列火柴中第 i 个火柴的高度，bi 表示第二列火柴中第 i 个火柴的高度。

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

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

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

输入格式

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

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

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

输出格式

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

数据范围

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

输入样例：

4
2 3 1 4
3 2 1 4

输出样例：

1

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

struct Node{
    int x,pos;
};
int n;
int tr[100001];
int c[100001];
struct Node a[100001],b[100001];
int lowbit(int x)
{
    return x & -x;
}
void update(int x, int c)  // 位置x加c
{
    for (int i = x; i <= n; i += lowbit(i)) tr[i] += c;
}
int query(int x)  // 返回前x个数的和
{
    int res = 0;
    for (int i = x; i; i -= lowbit(i)) res += tr[i];
    return res;
}
bool cmp(struct Node n1,struct Node n2)
{
    return n1.x<n2.x;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i].x);
        a[i].pos=i;
    }
    for(int i=1;i<=n;i++){
        scanf("%d",&b[i].x);
        b[i].pos=i;
    }
    sort(a+1,a+1+n,cmp);
    sort(b+1,b+1+n,cmp);
    for(int i=1;i<=n;i++){
        c[a[i].pos]=b[i].pos;
    }
    long long ans=0;
    for(int i=1;i<=n;i++){
        update(c[i],1);
        ans+=i-query(c[i]);
    }
    printf("%d",ans%99999997);
    
    return 0;
}