NOIP 提高组2013 火柴排队 （Vijos P1842）

描述

涵涵有两盒火柴，每盒装有 n 根火柴，每根火柴都有一个高度。现在将每盒中的火柴各自排成一列，同一列火柴的高度互不相同，两列火柴之间的距离定义为： ∑ i=1 n (a i −b i ) 2   ，其中 a i   表示第一列火柴中第 i 个火柴的高度， b i   表示第二列火柴中第 i 个火柴的高度。

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

格式

输入格式

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

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

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

输出格式

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

样例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

限制

每个测试点1s。

提示

样例1说明

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

样例2说明

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

数据范围

对于 10%的数据， 1 ≤ n ≤ 10；
对于 30%的数据，1 ≤ n ≤ 100；
对于 60%的数据，1 ≤ n ≤ 1,000；
对于 100%的数据，1 ≤ n ≤ 100,000，0 ≤火柴高度≤ 2^31 − 1。

易贪心得到，a1 < a2 < a3 <...<aN , 且 b1 < b2 < b3 <...<bN的情况下为最优解。

将两个数组分别离散到1~n的范围内便于统计，可将a数组不动且作为相对的大小，将b数组改为按照相对大小顺序排序，接着就是交换次数问题了，也就是求逆序对。

线段树实现。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <set>
#include <stack>
#include <cctype>
#include <algorithm>
#define lson o<<1, l, m
#define rson o<<1|1, m+1, r
using namespace std;
typedef __int64 LL;
const int maxn = 1000500;
const int mod = 99999997;
const int MAX = 0x3f3f3f3f;
int a, b, n, f[maxn], vis[maxn];
int num[maxn<<2];
struct C {
    int pos, num;
} ina[maxn], inb[maxn] ;
bool cmp(C x, C y) {
    return x.num < y.num;
}
void lisan(C *tt) {
    sort(tt+1, tt+1+n, cmp);
    for(int i = 1; i <= n; i++) tt[ tt[i].pos ].num = i;
}
void up(int o) {
    num[o] = num[o<<1] + num[o<<1|1];
}
void update(int o, int l, int r) {
    if(l == r) {
        num[o]++;
        return ;
    }
    int m = (l+r) >> 1;
    if(a <= m) update(lson);
    else update(rson);
    up(o);
}
int query(int o, int l, int r) {
    if(a <= l && r <= b) return num[o];
    int m = (l+r) >> 1;
    int res = 0;
    if(a <= m) res = (res + query(lson)) %mod;
    if(m < b ) res = (res + query(rson)) %mod;
    return res;
}
int main()
{
    cin >> n;
    for(int i = 1; i <= n; i++) { scanf("%d", &ina[i].num); ina[i].pos = i;}
    for(int i = 1; i <= n; i++) { scanf("%d", &inb[i].num); inb[i].pos = i;}
    lisan(ina);
    lisan(inb);
    for(int i = 1; i <= n; i++) vis[ ina[i].num ] = i;
    for(int i = 1; i <= n; i++) f[i] = vis[ inb[i].num ];

    b = n;
    int ans = 0;
    for(int i = 1; i <= n; i++) {
        a = f[i] + 1;
        ans = (ans%mod + query(1, 1, n)%mod) %mod;
        a = f[i];
        update(1, 1, n);
    }
    
    cout << ans << endl;
    return 0;
}