P1966 火柴排队

题面:https://www.luogu.org/problem/P1966

本题就是要将a序列第k大的元素和序列b中第k大的元素的位置换成一样的.
那么可以将a,b数组分别离散化,此时只要将a,b数组换成相同的即可.
令q[a[i]]=b[i],若a数组和b数组完全相同,则q[a[i]]=a[i],即q[i]=i.
那么就是要将q数组调成上升序列,那么求逆序对个数即为解.
注意:此处为归并排序求逆序对.

Code:
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=100002,M=99999997;
struct mt{
    int x,num;
}a[N],b[N];
int n,i,j,k,d[N],c[N],s;
bool cmp(mt p,mt q){
    if(p.x!=q.x)
        return p.x<q.x;
    return p.num<q.num;
}
void gb(int l,int r){
    if(l==r)
        return;
    int mid=l+r>>1;
    gb(l,mid);
    gb(mid+1,r);
    i=l;j=mid+1;k=0;
    while(i<=mid&&j<=r){
        if(c[i]<c[j])
            d[++k]=c[i++],s=(s+j-mid-1)%M;
        else
            d[++k]=c[j++];
    }
    while(i<=mid)
        d[++k]=c[i++],s=(s+j-mid-1)%M;
    while(j<=r)
        d[++k]=c[j++];
    for(i=1,j=l;i<=k;i++,j++)
        c[j]=d[i];
    return;
}
int main(){
    scanf("%d",&n);
    for(i=0;i<n;i++)
        scanf("%d",&a[i].x),a[i].num=i;
    for(i=0;i<n;i++)
        scanf("%d",&b[i].x),b[i].num=i;
    sort(a,a+n,cmp);    
    sort(b,b+n,cmp);
    for(i=0;i<n;i++)
        c[b[i].num]=a[i].num;
    gb(0,n-1);
    printf("%d",s);
    return 0;
}

转载于:https://www.cnblogs.com/ukcxrtjr/p/11521910.html