树状数组求逆序对;
对于一组任意顺序的数,记下他们的初位置,再进行排序,记下他们本来应该在的位置。然后,按最初位置从前往后,进行处理。
基本思想:
对于一个数要求它对应得逆序对,可以通过求前面小于等于它的数的个数,然后用当前已处理的数的个数-前面小于等于当前的数的个数得到。
如:
4 3 5 2 1
排序后 1 2 3 4 5
对于第一个数4,当前处理数的个数为1,本来应该在的位置是4,则在第4个位置上加1(利用树状数组的单点修改),然后用树状数组的区间查询,求出前4项的和1,用1-1=0。所以4对应得逆序数个数为0。
对于第二个数3,当前处理数的个数为2,本来应该在的位置是3,则在第3个位置上加1(利用树状数组的单点修改),然后用树状数组的区间查询,求出前3项的和1,用2-1=1。所以3对应得逆序数个数为1。
以此类推。
注意当数组中有重复的数字时,他们本来应该在的位置应记为同一个位置。
附上poj 2299 AC代码
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int N=5e5+5;
int n;
struct node
{
int val;
int num;
}b[N];
int a[N];
int tree[N];
bool cmp(node a,node b)
{
return a.val<b.val;
}
void add(int p,int x)//树状数组的应用
{
while(p<=n)
{
tree[p]+=x;
p+=p&(-p);
}
}
int read(int p)
{
int sum=0;
while(p>=1)
{
sum+=tree[p];
p-=p&(-p);
}
return sum;
}
int main()
{
while(scanf("%d",&n),n)
{
memset(tree,0,sizeof(tree));
memset(b,0,sizeof(b));
memset(a,0,sizeof(a));
for(int i=1;i<=n;i++)
{
scanf("%d",&b[i].val);
b[i].num=i;//记下原位置
}
sort(b+1,b+1+n,cmp);
a[b[1].num]=1;
int cnt=1;
for(int i=2;i<=n;i++)//元素的重复,顺序排列后应在同一位置
{
if(b[i].val!=b[i-1].val)
cnt++;
a[b[i].num]=cnt;
}
long long ans=0;
for(int i=1;i<=n;i++)
{
add(a[i],1);
ans+=(i-read(a[i]));
}
printf("%lld\n",ans);//用long long ,int 不够--WA!
}
return 0;
}