http://poj.org/problem?id=2299
最初做离散化的时候没太确定但是写完发现对的---因为后缀数组学的时候,,这种思维习惯了吧
1、初始化as[i]=i;对as数组按照num[]的大小间接排序
2、bs[as[i]]=i;现在bs数组就是num[]数组的离散化后的结果
3、注意,树状数组中lowbit(i) i是不可以为0的,0&(-0)=0,死循环...
#include <cstdio> #include <cstring> #include <algorithm> #include <string> #include <iostream> #include <cmath> #include <map> #include <set> #include <queue> using namespace std; #define ls(rt) rt*2 #define rs(rt) rt*2+1 #define ll long long #define ull unsigned long long #define rep(i,s,e) for(int i=s;i<e;i++) #define repe(i,s,e) for(int i=s;i<=e;i++) #define CL(a,b) memset(a,b,sizeof(a)) #define IN(s) freopen(s,"r",stdin) #define OUT(s) freopen(s,"w",stdout) const int MAXN = 500000 +100; int n; int C[MAXN],num[MAXN],as[MAXN],bs[MAXN],x[MAXN]; inline int lowbit(int i){return i&(-i);} void add(int i,int v) { for(;i<n;C[i]+=v,i+=lowbit(i));//printf("#i=%d#\n",i); } int sum(int i) { int s=0; for(;i>0;s+=C[i],i-=lowbit(i)); return s; } bool cmp(const int a,const int b) { return num[a]<num[b]; } int main() { //IN("poj2299.txt"); ll ans; while(~scanf("%d",&n) && n) { ans=0;CL(C,0);CL(x,0); for(int i=1;i<=n;i++) { scanf("%d",&num[i]); as[i]=i; } sort(as+1,as+1+n,cmp); for(int i=1;i<=n;i++) { bs[as[i]]=i; } ///////////////////// //for(int i=1;i<=n;i++) // printf("i=%d bs=%d\n",i,bs[i]); //////////////////////// for(int i=1;i<=n;i++) { add(bs[i],1); x[i]=sum(bs[i]-1); } for(int i=1;i<=n;i++) { ans+=sum(bs[i]-1)-x[i]; } printf("%lld\n",ans); } return 0; }