您的位置:首页 > 其它

POJ 2299 快速求逆序数

2013-11-01 20:45 302 查看
逆序数的本质就是看前面有几个数比你大,那么设置一个数组,每次插入一个数值,记录插入的数字个数,每次统计从1到插入的这个数之间的数字个数,这个数字是顺序数,用此时插入的数的总个数一减就是现在这个数列的逆序数。

统计从1到n的数的个数使用树状数组就可以了。

该死的,一开始没用64位整数,挂了。后来改成long long的时候又没有修改printf,修改成printf("%I64d",ans)就过掉了

#include<stdio.h>
#include<algorithm>
#include<utility>
#include<string.h>
using namespace std;

typedef long long ll;

typedef pair<ll,ll> P;

P num[600000];

ll b[600000];// discrete as origin sequence

ll bit[600000];

ll n;
ll sum(ll i)
{
ll s=0;
while(i>0)
{
s+=bit[i];
i-=i&(-i);
}
return s;
}

void add(ll i,ll x)
{
while(i<=n)
{
bit[i]+=x;
i+=i&(-i);
}
}

int main()
{
while(scanf("%d",&n)&&n!=0)
{
memset(bit,0,sizeof(bit));
for(ll i=1;i<=n;i++)
{
ll tn;
scanf("%d",&tn);
num[i]=P(tn,i);
}
sort(num+1,num+1+n);
for(ll i=1;i<=n;i++)
b[num[i].second]=i;
ll ans=0;
for(ll i=1;i<=n;i++)
{
add(b[i],1);
ans+=(i-sum(b[i]));
}
printf("%I64d\n",ans);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: