前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >poj 2299 Ultra-QuickSort 求逆序数 树状数组解法

poj 2299 Ultra-QuickSort 求逆序数 树状数组解法

作者头像
xindoo
发布2021-01-22 15:21:21
3540
发布2021-01-22 15:21:21
举报
文章被收录于专栏:XINDOO的专栏XINDOO的专栏

题目链接

逆序的概念大家都知道,一个数到逆序数就是该数左边大于它到数的个数。

很多没学过数据结构的人一上来肯定就是一个个数了,看看数据量500k,显然这种暴力的方法是行不通的。

我们换种想法,可以在输入过程中对每个数的逆序数求解,建一个vis数组(初始化为0),只要输入一个数,在它的位置标记为1,然后计算出它的左边一共有多少数被标记了就可以知道多少个数比他小了,当然逆序数也就知道了,求从左到右数的和,这是树状数组最擅长的了。

但我们看看每个数的范围是0 ≤ a[i] ≤ 999,999,999,我们不可能开那么大的数组,即使开的了也会浪费很多,这个时候我们就要对数据进行离散化处理了。

所谓离散化,我们的了解就是对原数组排序,然后用所在位置的下标代替原数,这样我们就可以把数据范围缩小到1-500000,这个数组是开的下的。

代码:

代码语言:javascript
复制
#include <stdio.h>
#include <algorithm>
#include <string.h>
using namespace std;
const int maxn = 500005;

struct number
{
    int a, b, pos;
}num[maxn];  //这个结构体是为离散化创建的,

int t[maxn];
int n;
int lowbit(int x)
{
    return x&(-x);
}

void update(int x)
{
    while (x <= n)
    {
        t[x] += 1;
        x += lowbit(x);
    }
}

int getsum(int x)
{
    int s = 0;
    while (x)
    {
        s += t[x];
        x -= lowbit(x);
    }
    return s;
}

int cmp(number x, number y)
{
    return x.a < y.a;
}

int cmp2(number x, number y)
{
    return x.pos < y.pos;
}

int main()
{
    while (scanf("%d", &n) && n)
    {
        for (int i = 1; i <= n; i++)
        {
            scanf("%d",&num[i].a);
            num[i].pos = i;
        }
        sort(num+1, num+1+n, cmp);
        for (int i = 1; i <= n; i++)
            num[i].b = i;
        sort(num+1, num+1+n, cmp2);  //两次的排序就是离散化的过程
        memset(t, 0, sizeof(t));
        __int64 sum = 0;
        for (int i = 1; i <= n; i++)
        {
            sum += (__int64)(i - 1 - getsum(num[i].b));
            update(num[i].b);
        }
        printf("%I64d\n",sum);
    }
    return 0;
}

关于这个题,还有归并排序的解法,虽然内存用的少了,但耗时增加了。

本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2013-04-25,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档