[NOIP2013][vijos1842]火柴排队（数学相关+离散化+bit）

题目描述

传送门

题解

首先把给出的式子化一化。
$\sum\limits_{i=1}^n (a_i-b_i)^2=\sum\limits_{i=1}^n (a_i^2+b_i^2-2a_ib_i)=\sum\limits_{i=1}^na_i+\sum\limits_{i=1}^nb_i-2\sum\limits_{i=1}^na_ib_i$
实际上也就是求$\sum\limits_{i=1}^na_ib_i$
假设$a_1>a_2,b_1>b_2$，那么$a_1b_1+a_2b_2-(a_1b_2+a_2b_1)=a_1(b_1-b_2)+a_2(b_2-b_1)>0$显然成立。
所以，当a序列和b序列第k大值相互对应的时候答案最优。
可知两个序列同时变和只变一个序列是等价的。
将两个序列离散化一下然后对应编号，变换次数就是逆序对数。

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define N 100005
#define Mod 99999997

int n,cnt,ans;
int a[N],b[N],p[N],q[N],renum[N],lsh[N];
int C[N];

int cmpa(int x,int y)
{
    return a[x]<a[y];
}
int cmpb(int x,int y)
{
    return b[x]<b[y];
}
void add(int loc)
{
    for (int i=loc;i<=n;i+=i&(-i))
        C[i]++;
}
int query(int loc)
{
    int ans=0;
    for (int i=loc;i>=1;i-=i&(-i))
        ans+=C[i];
    return ans;
}
int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;++i) scanf("%d",&a[i]),p[i]=i;
    for (int i=1;i<=n;++i) scanf("%d",&b[i]),q[i]=i;

    sort(p+1,p+n+1,cmpa);
    for (int i=1;i<=n;++i) renum[++cnt]=p[i];
    sort(q+1,q+n+1,cmpb);cnt=0;
    for (int i=1;i<=n;++i) lsh[q[i]]=renum[++cnt];

    for (int i=n;i>=1;--i)
    {
        ans=(ans+query(lsh[i]-1))%Mod;
        add(lsh[i]);
    }
    printf("%d\n",ans);
}

总结

①数学相关的式子上手化一化，不要瞪着眼睛看。