再清楚地打一边用树状数组求逆序对,记不住就只能多打,万事唯手熟耳。
这里都用的int,实际题目数据要求会不一样,必要时用long long int。这里的一步离散化的作用是如果数列中元素的大小比较大就不易直接开大小范围的树状数组来做,而是先按元素值排序然后比较的是元素原来的index,道理是一样的。
#include <iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;
class node//节点
{
public:
int value;
int index;
};
node nodes[100005];//输入数列
int a[100005];//离散化数组
int c[100005];//树状数组
int n;//数列长度
int result;//结果值
bool cmp(node a,node b)//按节点值比较,用来排序
{
if(a.value<b.value)
return true;
else return false;
}
int lowbit(int x)//lowbit函数
{
return x&(-x);
}
void update(int t,int value)//更新树状数组或叫插入元素,有地方叫change或add
{
for(int i=t;i<=n;i+=lowbit(i))
{
c[i]+=value;
}
}
int getsum(int x)//求前几项和,这里用来求x之前比x大的个数
{
int ans=0;
for(int i=x;i>=1;i-=lowbit(i))
{
ans+=c[i];
}
return ans;
}
int main()
{
scanf("%d",&n);
int tmp;
for(int i=1;i<=n;i++)
{
scanf("%d",&tmp);
nodes[i].value=tmp;
nodes[i].index=i;
}
sort(nodes+1,nodes+n+1,cmp);//按节点的value排序
for(int i=1;i<=n;i++)//离散化
{
a[nodes[i].index]=i;
}
result=0;
for(int i=1;i<=n;i++)
{
update(a[i],1);
result+=i-getsum(a[i]);//之前比a[i]大的数
}
printf("%d\n",result);
return 0;
}