题目:http://poj.org/problem?id=2299
题意:求逆序数。
分析:以前用归并写过,也可以用树状数组求,不过要离散化一下。简单说一下,对于每一个数b[i],求出它后面有多少个数比它小,就是查询[1,b[i]-1](离散化后)有多少个数存在就行了。更新,更新所有管辖b[i]的管辖点。
代码:
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 5e5+5;
int tree[maxn],a[maxn],b[maxn],n,tail;
int Find(int x)
{
int down=1,mid,up=tail;
while(down<=up)
{
mid=(down+up)>>1;
if(a[mid]>x)
up=mid-1;
else if(a[mid]==x)
return mid;
else
down=mid+1;
}
return mid;
}
int lowbit(int x)
{
return x&-x;
}
void update(int x,int v)
{
for(int i=x;i<=n;i+=lowbit(i))
tree[i]+=v;
}
int query(int x)
{
int ret(0);
for(int i=x;i>0;i-=lowbit(i))
ret+=tree[i];
return ret;
}
int main()
{
int i,temp;
while(scanf("%d",&n),n)
{
memset(tree,0,sizeof(tree[0])*(n+2));
for(i=1;i<=n;i++)
{
scanf("%d",&a[i]);
b[i]=a[i];
}
sort(a+1,a+n+1);
tail=unique(a+1,a+n+1)-(a+1);
long long ans=0;
for(i=n;i>=1;i--)
{
temp=Find(b[i]);
update(temp,1);
ans+=query(temp-1);
}
printf("%lld\n",ans);
}
return 0;
}
归并代码:
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 5e5+5;
int a[maxn],ta[maxn];
long long ans=0;
void merge_sort(int A[],int x,int y,int t[])
{
if(y>x)
{
int m=(x+y)>>1;
merge_sort(A,x,m,t);
merge_sort(A,m+1,y,t);
int p=x,q=m+1,i=x;
while(p<=m || q<=y)
{
if(q>y || p<=m && A[p]<=A[q])
t[i++]=A[p++];
else
{
t[i++]=A[q++];
ans+=q-i;
}
}
for(i=x;i<=y;i++)
A[i]=t[i];
}
}
int main()
{
int n;
while(scanf("%d",&n),ans=0,n)
{
for(int i=0;i<n;i++)
scanf("%d",&a[i]);
merge_sort(a,0,n-1,ta);
printf("%lld\n",ans);
}
return 0;
}