您的位置:首页 > 产品设计 > UI/UE

POJ 2299 -Ultra-QuickSort-树状数组求逆序数

2016-03-15 22:13 459 查看
POJ 2299 Ultra-QuickSort

使用树状数组记录逆序对数。

把数组按照大小顺序插入,getsum(i)就是i前面的比他大的数。

#include <cstdio>
#include <algorithm>
#include <cstring>

using namespace std;

const int maxn = 500005;
int reflect[maxn],c[maxn],N;

struct Node
{
int val;
int pos;
bool operator < (const Node &r) const
{
return val < r.val;
}
}node[maxn];

int lowbit(int x)
{
return x&(-x);
}

void update(int x)
{
while(x<=N)
{
c[x] += 1;
x += lowbit(x);
}
}

int getsum(int x)
{
int sum = 0;
while(x > 0)
{
sum += c[x];
x -= lowbit(x);
}
return sum;
}

int main()
{
while(~scanf("%d",&N) && N)
{
for(int i=1;i<=N;i++)
{
scanf("%d",&node[i].val);
node[i].pos = i;
}
sort(node+1,node + N+1);
for(int i=1;i<=N;i++) reflect[node[i].pos] = i;
memset(c,0,sizeof c);

long long ans = 0;
for(int i=1;i<=N;i++)
{
update(reflect[i]);
ans += i - getsum(reflect[i]);
}
printf("%lld\n",ans);
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: