【题解】洛谷P1966 [NOIP2013TG] 火柴排队(树状数组+逆序对)

mac2022-06-30  90

次元传送门:洛谷P1966

思路

显然在两排中 每排第i小的分别对应就可取得最小值(对此不给予证明懒)

所以我们只在意两排的火柴是第几根 高度只需要用来进行排序(先把两个序列改成有序的方便离散化)

因此我们对火柴的高度进行离散化 把火柴高度变为1到n的序列

然后我们只需要对一个序列a固定 求另一个序列b相对于前一个序列a的逆序对即可

举个栗子?:

a[]=1 123 7 966 1208; b[]=21 34 36 25 111;

离散之后得到:a[]=1 3 2 4 5;  b[]=1 3 4 2 5;

接下来是关键 对于每个a[i]的值对应一个i,b[i]映射到新序列的c[i]中 即c[a[i]]=b[i];(相当于以 a[i]为关键字对序列b[i]排序

解释:即a中1在b中的位置为1 a中2在b中的位置为4 a中3在b中的位置为2 a中4在b中的位置为3 a中5在b中的位置为5

得到:c[]=1 4 3 2 5

最后求c的逆序对个数即可(用树状数组)

这里用到正序做树状数组求逆序对(摘自此blog)

正序做树状数组,那么当前下标减掉当前数字的前缀和即为以该数为较小数的逆序对个数。

因为是正序,那么对于每个当前的数,已加入的数字个数(算当前数)即为当前数字在数列中的下标,也就是树状数组中已经加入了这么多个数,那么他的前缀和代表小于它且在他前面的数的个数,用总数减掉前缀和即为以该数为较小数的逆序对个数,同样,我们也不需要考虑多算或少算的情况发生。

代码

#include<iostream> #include<algorithm> using namespace std; #define maxn 100010 #define mod 99999997 struct March { int high; int order;//用来离散化 }a[maxn],b[maxn]; int n,ans; int c[maxn],e[maxn]; bool cmp(March a,March b) { return a.high<b.high; } int lowbit(int x) { return x&-x; } void add(int x,int t) { while(x<=n) { e[x]+=t; e[x]%=mod; x+=lowbit(x); } } int sum(int x) { int temp=0; while(x) { temp+=e[x]; temp%=mod; x-=lowbit(x); } return temp; } int main() { cin>>n; for(int i=1;i<=n;i++) { cin>>a[i].high; a[i].order=i; } for(int i=1;i<=n;i++) { cin>>b[i].high; b[i].order=i; } sort(1+a,1+a+n,cmp); sort(1+b,1+b+n,cmp); for(int i=1;i<=n;i++) c[a[i].order]=b[i].order;//关键 for(int i=1;i<=n;i++) { add(c[i],1);//离散后的顺序就是正确的顺序 ans+=i-sum(c[i]);//当前有几个元素减去之前比他大的数的个数 ans%=mod; } cout<<ans; }

 

转载于:https://www.cnblogs.com/BrokenString/p/9898640.html

最新回复(0)