树状数组 求 逆序对

在这里插入图片描述

在这里插入图片描述

  • 如何统计第i个数与1~第i - 1个数构成多少个逆序对呢?
  • 考虑根据值来建树状数组,初始树状数组为全0,先按序列从左到右将数据的值对应的位置的数加一,代表又有一个数出现。因此,在循环到第i项时,前i-1项已经加入到树状数组内了,树状数组内比ai大的都会与ai构成逆序对,产生的逆序对数量是i-sum(ai)
  • 根据ai来建树状数组空间不够诶?
  • 确实不够,但我们需要的只是元素之间的相对大小,这启发我们对数据离散化,先将数据排序,再用1~n分别对应n个数表示它们的相对大小,对新的序列建树状数组空间就够了(n≤5×10^5 )
  • 相等的元素是否会导致求解错误?每一个数(不管是否相等)对应的新数都不同诶?
  • 不处理的话会出错的,问题的关键在于是否有与ai相等的元素在ai前被加入且其相对大小标记更大。出现这种情况就会误将两个相等的数判为逆序对。怎么解决呢,只要所有与ai相等的元素中,先出现的标记也更小就好了(我们只统计相对更大的)。具体只需要在排序时将ai作为第一关键字,下标(第几个出现)作为第二关键字从小到大排序即可。
#include <iostream>
#include <algorithm>

using namespace std;

typedef long long ll;

const int N = 5e5 + 10;

int Rank[N], tr[N];
int n;

struct node
{
    int num, val;
}a[N];

bool cmp(node a, node b)
{
    if (a.val == b.val)
        return a.num < b.num;
    return a.val < b.val;
}

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

void add(int x, int c)
{
    for (int i = x; i <= n; i += lowbit(i)) tr[i] += c;
}

ll sum(int x)
{
    ll res = 0;
    for (int i = x; i; i -= lowbit(i)) res += tr[i];
    return res;
}

int main()
{
    scanf("%d", &n);
    
    for (int i = 1; i <= n; i ++ )
        scanf("%d", &a[i].val), a[i].num = i;
    
    sort(a + 1, a + n + 1, cmp);
    
    for (int i = 1; i <= n ; i ++ )
        Rank[a[i].num] = i;
    
    ll ans = 0;
    for (int i = 1; i <= n; i ++ )
    {
        add(Rank[i], 1);
        ans += i - sum(Rank[i]);
    }
    
    printf("%lld\n", ans);
    
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值