P1908 逆序对(树状数组)

题目传送门

题意: 给你一个数组,给定逆序对的定义:对于任意一对i,j,有i>j&&a[i]<a[j],问你这个数组有多少个逆序对。

数据范围: n<=5e5,a[i]<=1e9

思路: 我们用树状数组解此题。c[i]表示数字i出现了c[i]次,我们发现直接用树状数组更新之后,每次更新完这个点,就对这个点进行查询,查询之前有多少个点大于a[i],最后输出答案即可。但是我们发现a[i]是1e9级别的,我们数组开不了这么大,而逆序对的判定只和他们之间的相对大小有关,和数字的绝对大小无关,所以我们之前先把数组离散化处理之后再求值。

代码:

#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ls p<<1
#define rs p<<1|1
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define ll long long
#define int long long
#define pii pair<int,int>
#define ull unsigned long long
#define all(x) x.begin(),x.end()
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
char *fs,*ft,buf[1<<20];
#define gc() (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<20,stdin),fs==ft))?0:*fs++;
inline int read(){int x=0,f=1;char ch=gc();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=gc();}
return x*f;}
using namespace std;
const int N=1e6+5;
const int inf=0x3f3f3f3f;
const int mod=998244353;
const double eps=1e-20;
const double PI=acos(-1);
int a[N],b[N],c[N];
int lowbit(int x)
{
    return x&-x;
}
int sum(int x)
{
    int s=0;
    for(int i=x;i;i-=lowbit(i))
        s+=c[i];
    return s;
}
void add(int x,int k)
{
    for(int i=x;i<N;i+=lowbit(i))
        c[i]+=k;
}
signed main()
{
    int n=read();
    for(int i=1;i<=n;i++)
    {
        a[i]=read();
        b[i]=a[i];
    }
    sort(b+1,b+n+1);
    int m=unique(b+1,b+n+1)-b-1;
    for(int i=1;i<=n;i++)
    {
        a[i]=lower_bound(b+1,b+m+1,a[i])-b;
    }
    int res=0;
    for(int i=1;i<=n;i++)
    {
        add(a[i],1);
        res+=sum(N-1)-sum(a[i]);
    }
    cout<<res<<endl;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章