F - Shift and Inversions(树状数组求逆序对)

最新推荐文章于 2024-08-22 08:21:12 发布

原创最新推荐文章于 2024-08-22 08:21:12 发布 · 199 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了如何使用树状数组（前缀和）解决n个由0到n-1的数组逆序对问题。代码展示了从输入数组中计算逆序对数量的方法，通过动态更新和查询操作，提高了查找效率。

题目链接
题意：求n个如下由0~n-1组成的数组的逆序对
a1,a2,…,an
a2,a3,…,a1
a3,a4,…a2

#include<bits/stdc++.h>
using namespace std;
const int N=3e5+10;
const int inf=0x3f3f3f3f;
#define ll long long
int n,a[N],c[N];

void add(int x){
    for(int i=x;i<=n;i+=i&-i)
        c[i]++;
}

int query(int x){
    int ans=0;
    for(int i=x;i>0;i-=i&-i){
        ans+=c[i];
    }
    return ans;
}

int main(){
    cin>>n;
    ll ans=0;
    for(int i=1;i<=n;i++){
        cin>>a[i];a[i]++;//注意树状数组是从1开始的，而题目给的序列是0~n-1
        add(a[i]);
        ans+=i-query(a[i]);
        //cout<<query(a[i])<<endl;
        //query求出的是小于等于a[i]的个数，i是现有的个数
    }
    for(int i=1;i<=n;i++){
        cout<<ans<<endl;
        ans+=n-a[i]-a[i]+1;
    }

return 0;
}