POJ 2299 - Ultra-QuickSort(BIT)

题目链接 https://cn.vjudge.net/problem/POJ-2299

【题意】
给你一组长度为 N N 的序列 A N < = 5 × 1 0 5 , A i < = 1 0 9 A(N<=5×10^5,A_i<=10^9) ,求出序列中逆序对的个数

【思路】
先对序列中的元素进行离散化,然后用树状数组去优化即可,或者利用归并排序

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=500005;

int n;
int a[maxn],c[maxn];
int bit[maxn];

void add(int i,int x){
    while(i<=n){
        bit[i]+=x;
        i+=i&-i;
    }
}

int sum(int i){
    int ans=0;
    while(i>0){
        ans+=bit[i];
        i-=i&-i;
    }
    return ans;
}

int main(){
    while(scanf("%d",&n)==1 && n){
        for(int i=0;i<n;++i){
            scanf("%d",&a[i]);
            c[i]=a[i];
        }
        sort(c,c+n);
        for(int i=0;i<n;++i){
            a[i]=lower_bound(c,c+n,a[i])-c+1;
        }
        memset(bit,0,sizeof(bit));
        ll ans=0;
        for(int i=0;i<n;++i){
            ans+=sum(n)-sum(a[i]);
            add(a[i],1);
        }
        printf("%lld\n",ans);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/xiao_k666/article/details/82782735
今日推荐