树状数组求逆序对数小白讲解+poj2299

求逆序对数:最进刚刚学习的树状数组,一直感觉没啥用处,没想到根据树状数组的特殊结构求逆序对数还是挺方便的

题目链接

题意就是求逆序对:
给了n个数,但是数比较分散,所以我们需要用离散化一下,把分散的数转化为为紧凑而且好求的数,然后排一下序,之后求逆序数即可

题解:由于给的数比较散,而且我们求逆序数只需要根据大小来判断逆序对数,所以把分散的数离散化一下,然后根据树状数组求一下逆序数即可

如果不理解树状数组求逆序数可以参考:
大神博客
AC代码:

#include
#include
#include
using namespace std;
#define ll long long
#define maxn 500005
ll a[maxn];
ll c[maxn];
struct node
{
    ll val,pos;
} temp[maxn];
bool cmp(node x,node y)
{
    return x.val<y.val;
}
ll n;
ll lowbit(ll x)
{
    return x&(-x);
}
void update(ll i,ll k)
{
    while(i<=n)
    {
        c[i]+=k;
        i+=lowbit(i);
    }
}
ll getsum(ll i)
{
    ll ans=0;
    while(i>0)
    {
        ans+=c[i];
        i-=lowbit(i);
    }
    return ans;
}
int main()
{
    while(~scanf("%lld",&n)&&n)
    {
        for(int i=1;i<=n;i++)
            c[i]=0;
        for(ll i=1; i<=n; i++)
        {
            scanf("%lld",&temp[i].val);
            temp[i].pos=i;
        }
        sort(temp+1,temp+1+n,cmp);
        ll ans=0;
        for(ll i=1; i<=n; i++)
        {
            update(temp[i].pos,1);
            ans+=getsum(n)-getsum(temp[i].pos);
        }
        printf("%lld\n",ans);
    }
}

你可能感兴趣的:(ACM刷题题解,数据结构)