arc120C - Swaps 2

传送门

通过观察两个数列的关系 发现：
令a[i]=a[i]+i 、b[i]=b[i]+i;若新的a[i]与新的b[i]构成的元素相同 这时a才能经过变换形成b
如果对应元素的位置不同 就要想办法给它移动过去。可以给a赋上对应的b的位置的序号。因为要一个一个交换位置 想一想就发现是个逆序对问题 数据量2e5,所以用树状数组来解决

//cyc
#pragma GCC optimize("Ofast")
#pragma GCC target("avx,avx2,fma")
#pragma GCC optimization ("unroll-loops")
#include<bits/stdc++.h>
#define rep(i,a,n) for(int i=a;i<=n;i++)
#define per(i,a,n) for(int i=n;i>=a;i--)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define mst(a) memset(a,0,sizeof a)
#define int long long
using namespace std;
typedef pair<int,int> pii;
const int maxn=2e5+5;
struct rec
{
    int val,pos;
}a[maxn],b[maxn];
int d[maxn];
int n;
bool cmp(rec r1,rec r2)
{
    if(r1.val!=r2.val)return r1.val<r2.val;
    else return r1.pos<r2.pos;
}
int c[maxn];
int lowbit(int x){
    return x&(-x);
}
void updata(int i,int k){    //在i位置加上k
    while(i <= n){
        c[i] += k;
        i += lowbit(i);
    }
}
int getsum(int i){        //求A[1 - i]的和
    int res = 0;
    while(i > 0){
        res += c[i];
        i -= lowbit(i);
    }
    return res;
}
signed main()
{
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    cin>>n;
    int k;
    for(int i=1;i<=n;i++){
        cin>>k;
        k+=i;
        a[i]={k,i};
    }
    int flag=0;
    for(int i=1;i<=n;i++){
        cin>>k;
        k+=i;
        b[i]={k,i};
    }
    sort(a+1,a+1+n,cmp);
    sort(b+1,b+1+n,cmp);
    for(int i=1;i<=n;i++){
        if(a[i].val!=b[i].val){
            flag=1;
            break;
        }
        d[a[i].pos]=b[i].pos;
    }
    if(flag){
        cout<<-1<<endl;
    }
    else{
        int ans=0;
        for(int i=1;i<=n;i++){

            ans+=(getsum(n)-getsum(d[i]));
            updata(d[i],1);
        }
        cout<<ans<<endl;
    }
}