poj 2299 树状数组求逆序数+离散化

http://poj.org/problem?id=2299

最初做离散化的时候没太确定但是写完发现对的---因为后缀数组学的时候,,这种思维习惯了吧

1、初始化as[i]=i;对as数组按照num[]的大小间接排序

2、bs[as[i]]=i;现在bs数组就是num[]数组的离散化后的结果

3、注意,树状数组中lowbit(i)  i是不可以为0的,0&(-0)=0,死循环...


#include <cstdio>
#include <cstring>
#include <algorithm>
#include <string>
#include <iostream>
#include <cmath>
#include <map>
#include <set>
#include <queue>
using namespace std;

#define ls(rt) rt*2
#define rs(rt) rt*2+1
#define ll long long
#define ull unsigned long long
#define rep(i,s,e) for(int i=s;i<e;i++)
#define repe(i,s,e) for(int i=s;i<=e;i++)
#define CL(a,b) memset(a,b,sizeof(a))
#define IN(s) freopen(s,"r",stdin)
#define OUT(s) freopen(s,"w",stdout)

const int MAXN = 500000 +100;
int n;

int C[MAXN],num[MAXN],as[MAXN],bs[MAXN],x[MAXN];

inline int lowbit(int i){return i&(-i);}
void add(int i,int v)
{
    for(;i<n;C[i]+=v,i+=lowbit(i));//printf("#i=%d#\n",i);
}
int sum(int i)
{
    int s=0;
    for(;i>0;s+=C[i],i-=lowbit(i));
    return s;
}

bool cmp(const int a,const int b)
{
    return num[a]<num[b];
}

int main()
{
    //IN("poj2299.txt");
    ll ans;
    while(~scanf("%d",&n) && n)
    {
        ans=0;CL(C,0);CL(x,0);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&num[i]);
            as[i]=i;
        }
        sort(as+1,as+1+n,cmp);
        for(int i=1;i<=n;i++)
        {
            bs[as[i]]=i;
        }
        /////////////////////
        //for(int i=1;i<=n;i++)
          //  printf("i=%d bs=%d\n",i,bs[i]);
        ////////////////////////
        for(int i=1;i<=n;i++)
        {
            add(bs[i],1);
            x[i]=sum(bs[i]-1);
        }
        for(int i=1;i<=n;i++)
        {
            ans+=sum(bs[i]-1)-x[i];
        }
        printf("%lld\n",ans);
    }
    return 0;
}


你可能感兴趣的:(poj 2299 树状数组求逆序数+离散化)