POJ 2299-Ultra-QuickSort （树状数组+离散化）

来源：互联网发布：淘宝送一个月优酷会员编辑：程序博客网时间：2024/05/18 09:20

题目：http://poj.org/problem?id=2299

思路：树状数组求和，求逆序数对。

CODE：

#include <iostream>#include <stdio.h>#include <algorithm>const int M=500005;using namespace std;typedef long long ll;int a[M],sa[M];int c[M];bool cmp(int x,int y){    return x<y;}int lowbit(int xx){    return xx&(-xx);}int sum(int x){    int s=0;    while(x > 0)    {        s+=c[x];        x-=lowbit(x);    }    return s;}void add(int x,int p,int NN){    while(x <= NN)    {        c[x]+=p;        x += lowbit(x);    }}int main(){    //freopen("in.in","r",stdin);    int N;    while(~scanf("%d",&N),N)    {        fill(c,c+N+1,0);        fill(a,a+N+1,0);        fill(sa,sa+N+1,0);        for(int i=1;i<=N;i++)        {            scanf("%d",&a[i]);            sa[i]=a[i];        }        sort(sa,sa+N+1,cmp);        ll ans=0;        for(int i=1;i<=N;i++)        {            int pl=lower_bound(sa+1,sa+N+1,a[i])-sa;            add(pl,1,N);            ans +=(ll)(sum(N) - sum(pl));        }        printf("%lld\n",ans);    }    return 0;}

0 0