poj 2299 Ultra-QuickSort 树状数组

田兵 — Wed, 16 Mar 2011 12:49:00 GMT

求逆序�Ҏ��Q�树状数�l?br>
数据范围较大�Q�要��L��化�?br>
�l�每一个数据一个id�Q?�W�i个数据的id为i�?然后从小到大排序�Q�对于每个id�?ans += read(n) - read(array[i].id)�Q�read(n) - read(array[i].id)表示原来在当前数的后面（其id大于当前数的id�Q�，
现在在当前数前面的数个数�Q�也��是逆序�Ҏ��?/span>

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int MAXVAL = 500005;

int tree[MAXVAL] ;
struct Type
{
    int num, id;
};

int n;
Type array[MAXVAL];

void update(int idx, int inc)  //更新idx的频�?/span>
{
    while(idx <= n)
    {
        tree[idx] += inc;
        idx += (idx & - idx);
    }
}

int read(int idx)   //��d��1--idx的频率和
{
    int sum = 0;
    while(idx > 0)
    {
        sum += tree[idx];
        idx -= (idx & - idx);
    }
    return sum;
}

int readSingle(int idx) // ��d��某个位置的频率， O(lg MAXVAL)
{
     int sum = tree[idx];
     if(idx > 0)
     {
         int z = idx - ( idx & - idx);  
         
         idx --;

         while( idx != z)
         {
              sum -= tree[idx];

              idx -= (idx & - idx);
         }
     }

     return sum;
}


bool cmp(const  Type &a, const Type &b)
{
    return a.num < b.num;
}
int main()
{
    while (scanf("%d",&n)  && n != 0)    
    {
        memset(array, 0, sizeof (array));
        memset(tree, 0, sizeof tree);

        // read the data
        for(int i = 1; i <= n; i ++)
        {
            scanf("%d",&array[i].num);
            array[i].id = i;
        }
    
        sort(array + 1, array + 1 + n, cmp);

        long long ans = 0;
        for(int i = 1; i <= n; i ++)
        {
            //printf( "cal   %d \n",read(n) - read(array[i].id));
            ans += read(n) - read(array[i].id);
            update(  array[i].id, 1);
        }
            
        cout << ans << endl;
    }


    return 0;
}

田兵 2011-03-16 20:49 发表评论

久久九九久精品国产,日本精品久久久久久久久免费,久久婷婷五月综合97色直播

poj 2299 Ultra-QuickSort 树状数组