ACM PKU 2299 Ultra-QuickSort
题目链接:http://poj.org/problem?id=2299
这道题是求逆序数的,离散化+树状数组;
注意的地方:1、离散化的比较函数;2、结果会爆int,所以必须用__int64或者是long long 类型来存储结果;
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
const int MAXN=500000;
int n_case;
int B[MAXN+10],C[MAXN+10];
struct node
{
int index,val;
} A[MAXN+10];
int cmp(const void * a,const void * b) //离散化比较函数模板;这里解决了一些相同的情况
{
return ((node *)a)->val - ((node *)b) ->val;
}
int lowbit(int a) //树状数组的关键地方;
{
return (-a)&a;
}
void insert(int x)
{
for(int i=x; i<MAXN; i+=lowbit(i))
{
C[i]++;
}
}
int getSum(int x)
{
int ans=0;
for(int i=x; i>0; i-=lowbit(i)) ans+=C[i];
return ans;
}
int main()
{
//freopen("in.txt","r",stdin);
long long sum; //用long long存储计算结果;
while(scanf("%d",&n_case)&&n_case!=0)
{
int temp;
sum=0;
for(int i=1; i<=n_case; i++)
{
scanf("%d",&temp);
A[i].val=temp;
A[i].index=i;
}
qsort(A+1,n_case,sizeof(A[0]),cmp);
for(int i=1; i<=n_case; i++)
{
B[A[i].index]=i;
}
memset(C,0,sizeof(C));
for(int i=n_case; i>0; i--) //这里从最后一个开始,查找放进树状数组C里比当前元素小的元素,即 前面所有之和;因为树状数组存储的是逆序数;
{
sum+=getSum(B[i]);
insert(B[i]);
}
printf("%I64d\n",sum);
}
return 0;
}
浙公网安备 33010602011771号