HDU 2689 Sort it(逆序对-BIT)

Description
给出一个长度为n的序列,输出其逆序对数
Input
多组输入,每组用例第一行为一整数n表示序列长度,第二行n个整数表示该序列,以文件尾结束输入
Output
对于每组用例,输出其逆序对数
Sample Input
3
1 2 3
4
4 3 2 1
Sample Output
0
6
Solution
树状数组求逆序对
Code

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<vector>
#include<string>
using namespace std;
#define maxn 1111
int num[maxn];
int bit[maxn];
struct node
{
    int val,pos;
}a[maxn];
int cmp(node x,node y)
{
    if(x.val!=y.val)return x.val<y.val;
    return x.pos<y.pos;
}
void init()//初始化 
{
    memset(bit,0,sizeof(bit));
} 
int lowbit(int x)//求lowbit 
{
    return x&(-x);
}
void update(int x,int v)//单点更新 
{
    while(x<=maxn)
    {
        bit[x]+=v;
        x+=lowbit(x);
    }
}
int getsum(int x) 
{
    int ans=0;
    while(x>0)
    {
        ans+=bit[x];
        x-=lowbit(x);
    }
    return ans; 
}
int solve(int num[],int n)//num表示要求逆序对的序列(从0开始),n表示序列长度 
{
    init();
    for(int i=1;i<=n;i++)
    {
        a[i].val=num[i-1];
        a[i].pos=i;
    }
    sort(a+1,a+n+1,cmp);
    int ans=0; 
    for(int i=n;i>0;i--)
    {
        ans+=getsum(a[i].pos);
        update(a[i].pos,1);
    }
    return ans;
}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        for(int i=0;i<n;i++)    
            scanf("%d",&num[i]);
        int ans=solve(num,n);
        printf("%d\n",ans);
    }
    return 0;
}

你可能感兴趣的:(HDU 2689 Sort it(逆序对-BIT))