备战蓝桥杯与天梯赛的同时复习一遍树状数组,刷完本题给我的启发很大,特此题解记录一下我的收获。
题目链接:https://www.luogu.com.cn/problem/P1966
题目描述:
涵涵有两盒火柴,每盒装有 n 根火柴,每根火柴都有一个高度。 现在将每盒中的火柴各自排成一列, 同一列火柴的高度互不相同, 两列火柴之间的距离定义为:∑∑(ai^2−bi^2)。
其中 ai 表示第一列火柴中第 i 个火柴的高度,bi 表示第二列火柴中第 i 个火柴的高度。
每列火柴中相邻两根火柴的位置都可以交换,请你通过交换使得两列火柴之间的距离最小。请问得到这个最小的距离,最少需要交换多少次?如果这个数字太大,请输出这个最小交换次数对 10^8−3取模的结果
刷的题目比较多的同学就会发现这道题的母题其实是用树状数组求逆序对,附上链接:
P1908 逆序对 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
简要复习一下思路,大概是通过在该i位前记录比大的数,因为在我们只有每次与相邻的数进行交换直到交换到不出现逆序对的位置为止,举个例子吧:
1 ,2 ,4,5 ,3 --> 1 ,2 ,4,3 ,5 --> 1 ,2 ,3,4 ,5
3会一直与前面比他大的两数(4,5)交换直到前面没有比他大的数为止,不难发现此时交换次数为,其中是在i位置的映射,而记录的是比i前比大的数的个数,这是逆序对的原理,回到本题,其实本题与其类似,我们要想最小化,最佳的办法是把中最小与最大的对应次小与次大的对应相减,由于数据太大我们需要离散化使其数值大小相对应,这里我采用的是坐标的方法离散化,用记录在i坐标下的映射,之后再对排序以便建立用记录对应的大小,最后是正常的逆序对操作:
#include<map>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
using ll=long long ;
using pii=pair<int,int>;
const int N=1e5+10;
const int mod=99999997;
int n;
int tr[N];
pii a[N],b[N];
int ra[N],rb[N];
int lowbit(int x)
{
return x&-x;
}
void add(int x,int v)
{
for(int i=x;i<N;i+=lowbit(i))
tr[i]+=v;
}
ll sum(int x)
{
ll res=0;
for(int i=x;i>=1;i-=lowbit(i))
res+=tr[i];
return res;
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
int x;
cin>>x;
a[i]={x,i};
}
for(int i=1;i<=n;i++)
{
int x;
cin>>x;
b[i]={x,i};
ra[a[i].second]=i;
}
sort(a+1,a+1+n);
sort(b+1,b+1+n);
for(int i=1;i<=n;i++)
rb[b[i].second]=a[i].second;
int ans=0;
for(int i=1;i<=n;i++)
{
add(ra[rb[i]],1);
ans=(ans%mod+(i-sum(ra[rb[i]]))%mod)%mod;
}
cout<<ans<<endl;
}