题意
给定一个长度为n的整数数列,请你计算数列中的逆序对的数量。
逆序对的定义如下:对于数列的第 i 个和第 j 个元素,如果满足 i < j 且 a[i] > a[j],则其为一个逆序对;否则不是。
输入格式
第一行包含整数n,表示数列的长度。
第二行包含 n 个整数,表示整个数列。
输出格式
输出一个整数,表示逆序对的个数。
数据范围
1≤n≤100000
输入样例:
6
2 3 4 5 6 1
输出样例:
5
思路
经典求逆序对问题,可以用归并排序来解
也可以用树状数组来解
复杂度都是
n
log
2
n
n\log_2n
nlog2n
但是树状数组显然更好写
而且归并排序复杂度可能到O(n2)
所以树状数组是求解逆序对的不二之选
算法思路
因为数据范围n最大是1e5,所以可以通过离散化来减少空间消耗。
然后是树状数组的增加操作和询问操作来求出每一个数前面有多少比自己大的数
大概思路是这样的
代码
#include<iostream>
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxn=500001;
int c[maxn];
struct Node
{
int v,index;
bool operator < (const Node &b) const
{
return v<b.v; //从小到大排序
}
}node[maxn];
int n,b[maxn];
void add(int i)
{
while(i<=n)
{
c[i]++;
i+=i&(-i);
}
}
long long sum(int i)
{
long long res=0;
while(i>0)
{
res+=c[i];
i-=i&(-i);
}
return res;
}
signed main()
{
cin>>n;
int a;
for(int i=1;i<=n;i++)
{
scanf("%d",&a);
node[i].index=i;
node[i].v=a;
}
sort(node+1,node+1+n);
long long ans=0;
int cnt=1;
b[node[1].index]=1;
for(int i=2;i<=n;i++){
if(node[i].v!=node[i-1].v){
cnt++;
}
b[node[i].index]=cnt;
}
for(int i=1;i<=n;i++)
{
add(b[i]); //离散化结果—— 下标等效于数值
ans+=i-sum(b[i]); //得到之前有多少个比你大的数(逆序对)
}
cout<<ans;
return 0;
}
当然,代码一部分借鉴了网上其他博客