牛客小白G-小苯的逆序对(树状数组+DP/莫比乌斯函数)

时间复杂度均为 O(nlognlogn)

DP做法

前置知识:

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define all(x) x.begin(),x.end()
#define no cout<<"No"<<endl
#define yes cout<<"Yes"<<endl
#define endl '\n'
// #define x first
// #define y second
typedef pair<int,int> PII;
const int N=200010;
const int mod=998244353;
const int INF=0x3f3f3f3f3f3f3f3f;
struct BIT {
    #define lowbit(x) (x&-x)
    int n;
    vector<int>tr;
    BIT(int size){
        n = size;
        tr.resize(size + 1, 0);
    }
    void add(int t,int val){
        for(int i=t;i<=n;i+=lowbit(i)){
            tr[i]+=val;
        }
    }
    int query(int t){
        int ret=0;
        for(int i=t;i;i-=lowbit(i)){
            ret+=tr[i];
        }
        return ret;
    }
};
vector<int>divs[N];
void get_divisor(int n=N-1){
    for(int i=1;i<=n;i++){
        for(int j=i;j<=n;j+=i){
            divs[j].push_back(i);
        }
    }
}

void solve(){
    int n;cin>>n;
    get_divisor(n);
    vector<int>a(n+1);
    vector<int>v[n+1];
    for(int i=1;i<=n;i++){
        cin>>a[i];
        for(auto j:divs[a[i]]){
            v[j].push_back(a[i]);
        }
    }
    int ans=0;
    BIT tr(n);
    vector<int>dp(n+1);
    for(int i=n/2;i>=1;i--){//这个排列的最大公因数<=n/2所以从n/2开始枚举即可
        for(int j=v[i].size()-1;j>=0;j--){
            tr.add(v[i][j],1);
            dp[i]+=tr.query(v[i][j]-1);
        }
        for(int j=v[i].size()-1;j>=0;j--)
            tr.add(v[i][j],-1);

        for(int j=i+i;j<=n;j+=i){
            dp[i]-=dp[j];
        }
    }

    cout<<dp[1]<<endl;
}
signed main(){
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    int _=1;
    while(_--)solve();
    return 0;
}

莫比乌斯函数做法

求出gcd(ai, aj)是1的倍数的逆序对个数,gcd(ai, aj)是2的倍数.... 是n的倍数

则ans=gcd[1]+gcd[2]+gcd[3]-gcd[6].... = mobius[i]*gcd[i] (gcd[i]表示是i的倍数的逆序对个数)

逆序对用树状数组求

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define all(x) x.begin(),x.end()
#define no cout<<"No"<<endl
#define yes cout<<"Yes"<<endl
#define endl '\n'
// #define x first
// #define y second
typedef pair<int,int> PII;
const int N=200010;
const int mod=998244353;
const int INF=0x3f3f3f3f3f3f3f3f;
struct BIT {
    #define lowbit(x) (x&-x)
    int n;
    vector<int>tr;
    BIT(int size){
        n = size;
        tr.resize(size + 1, 0);
    }
    void add(int t,int val){
        for(int i=t;i<=n;i+=lowbit(i)){
            tr[i]+=val;
        }
    }
    int query(int t){
        int ret=0;
        for(int i=t;i;i-=lowbit(i)){
            ret+=tr[i];
        }
        return ret;
    }
};

int mo[N];
int book[N];
int prime[N],cnt;
void mobius(int n=N-1){
    mo[1]=1;
    for(int i=2;i<=n;i++){//枚举k
        if(!book[i]){
            prime[cnt++]=i;
            mo[i]=-1;//质数的mo=-1
        }
        for(int j=0;prime[j]*i<=n;j++){//p*k<=n
            book[prime[j]*i]=1;
            if(i%prime[j]==0){//其质因数至少包含两个prime[j],故mo=0
                mo[prime[j]*i]=0;
                break;
            }
            mo[prime[j]*i]=-mo[i];//质因数互不相同
        }
    }
}
int mp[N];
void solve(){
    int n;cin>>n;
    mobius(n);
    vector<int>a(n+1);
    for(int i=1;i<=n;i++){
        cin>>a[i];
        mp[a[i]]=i;
    }
    int ans=0;
    BIT tr(n);
    for(int i=1;i<=n;i++)if(mo[i]!=0)
    {
        vector<int>v;
        for(int j=i;j<=n;j+=i){
            v.push_back(j);
        }
        sort(all(v),[&](int a,int b){
            return mp[a]>mp[b];
        });
        for(auto j:v){
            tr.add(j,1);
            ans+=mo[i]*tr.query(j-1);
        }
        for(auto j:v)tr.add(j,-1);
    }
    cout<<ans<<endl;
}
signed main(){
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    int _=1;
    while(_--)solve();
    return 0;
}

  • 3
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值