POJ 2299 Ultra-QuickSort (树状数组 求逆序对)

POJ 2299

题意:给你一组数字,能交换相邻元素,问需要交换多少次才能使其变成升序序列。

题解:实际上就是求逆序对个数,不过因为数据比较大,所以需要离散化一下。

#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstring>
#define fio ios::sync_with_stdio(false);cin.tie(0)
#define Pii pair<int,int> using namespace std; const double pi=acos(-1.0); const int N=5e6+5; using namespace std; int n; Pii p[N]; int a[N]; int c[N]; void add(int k,int num){ while(k<=n){ c[k]+=num; k+=k&-k; } } int getsum(int x){ int sum=0; while(x>0){ sum+=c[x]; x-=x&-x; } return sum; } int main(){ fio; while(cin>>n&&n){ memset(p,0,sizeof(p)); memset(c,0,sizeof(c)); for(int i=1;i<=n;i++){ cin>>p[i].fi; p[i].se=i;//下标 } sort(p+1,p+n+1); for(int i=1;i<=n;i++){ a[p[i].se]=i;//离散化 } ll ans=0; for(int i=1;i<=n;i++){ add(a[i],1); ans+=i-getsum(a[i]); } cout<<ans<<endl; } return 0; }

 

posted @ 2018-02-02 13:41  采蘑菇的小西佬  阅读(120)  评论(1编辑  收藏  举报