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

POJ 2299Ultra-QuickSort

使用树状数组记录逆序对数。

把数组按照大小顺序插入，getsum(i)就是i前面的比他大的数。

 #include <cstdio>
 #include <algorithm>
 #include <cstring>

 using namespace std;

 const int maxn = ;
 int reflect[maxn],c[maxn],N;

 struct Node
 {
     int val;
     int pos;
     bool operator < (const Node &r) const
     {
         return val < r.val;
     }
 }node[maxn];

 int lowbit(int x)
 {
     return x&(-x);
 }

 void update(int x)
 {
     while(x<=N)
     {
         c[x] += ;
         x += lowbit(x);
     }
 }

 int getsum(int x)
 {
     int sum = ;
     while(x > )
     {
         sum += c[x];
         x -= lowbit(x);
     }
     return sum;
 }

 int main()
 {
     while(~scanf("%d",&N) && N)
     {
         for(int i=;i<=N;i++)
         {
             scanf("%d",&node[i].val);
             node[i].pos = i;
         }
         sort(node+,node + N+);
         for(int i=;i<=N;i++) reflect[node[i].pos] = i;
         memset(c,,sizeof c);

         long long ans = ;
         for(int i=;i<=N;i++)
         {
             update(reflect[i]);
             ans += i - getsum(reflect[i]);
         }
         printf("%lld\n",ans);
     }
 }