归并或树状数组:逆序对的数量

在这里插入图片描述

1. 归并排序

参考链接:https://www.acwing.com/solution/content/5508/
在这里插入图片描述
在这里插入图片描述
左边有序,右边也有序。很自然的想到使用归并排序。

#include<iostream>
#include<cstdio>

using namespace std;

typedef long long LL;

const int N=100010;

int n,q[N],tmp[N];

//假设这个函数可以返回l到r区间内逆序对的个数
LL merge_sort(int q[],int l,int r)
{
    if(l>=r) return 0;
    
    int mid=l+r>>1;
    
    LL res=merge_sort(q,l,mid)+merge_sort(q,mid+1,r);
    int k=0,i=l,j=mid+1;
    
    while(i<=mid&&j<=r)
    {
        if(q[i]<=q[j]) tmp[k++]=q[i++];//这里的等号必须加,因为相等的话是不算的情况。
        else
        {
            tmp[k++]=q[j++];
            // 如果q[i] > q[j]了,那么q[i~mid]肯定都>q[j]
            // i~mid 一共mid-i+1个数。
            res+=mid-i+1; 
        }
    }
    
    while(i<=mid) tmp[k++]=q[i++];
    while(j<=r) tmp[k++]=q[j++];
    
    for(i=l,j=0;i<=r;i++,j++) q[i]=tmp[j];
    return res;
}

int main()
{
    cin>>n;
    for(int i=0;i<n;i++) cin>>q[i];
    
    cout<<merge_sort(q,0,n-1)<<endl;
    
    return 0;
}

2. 离散化+树状数组

一下解法来自:https://www.acwing.com/solution/content/12233/

在这里插入图片描述

// 我还是没懂
#include<cstdio>
#include<algorithm>

using namespace std;

typedef long long LL;

const int N = 500010;

int n, m;
// 数最大到1e9,但是最多只有10000个数
// 因此需要离散化, a是原数组,b是离散化之后的数组
int a[N], b[N];

// 二分查找
// find(a[i])返回整数 a[i] 在 b数组中的下标
int find(int x)
{
    int l = 1, r = m;
    while(l < r)
    {
        int mid = l + r >> 1;
        if(b[mid] >= x)  r = mid;
        else l = mid + 1;
    }
    return l;
}

int tr[N];

void add(int x)
{
    for(int i = x; i <= m; i += i & -i)  tr[i] ++;
}

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

int main()
{
    scanf("%d", &n);
    
    for(int i = 1; i <= n; i ++)  scanf("%d", &a[i]),  b[i] = a[i];

    // 先对b进行排序
    sort(b + 1, b + n + 1);

    // 再对b进行去重
    // 使 b[1] 到 b[m]b是严格单调递增的整数序列,并包含了 a 数组中的每一个数
    m = 1;
    for(int i = 2; i <= n; i ++)
        if(b[i] != b[m])  b[++ m] = b[i];

    // 然后从前往后遍历 a 数组,树状数组维护 ≤b[i] 的数有几个
    // 对于每一个 a[i]我们需要知道前面有几个数大于它
    // 可以用前面的数的个数减去 ≤a[i] 的数的个数,累加到答案里
    // res += sum(m) - sum(find(a[i]))
    // 然后再往树状数组里加一个 a[i] 就可以了
    LL res = 0;
    for(int i = 1; i <= n; i ++)
    {
        // find(a[i]) 返回整数 a[i] 在 b 数组中的下标
        int x = find(a[i]);
        res += sum(m) - sum(x);
        add(x);
    }
    printf("%lld", res);
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
树状数组(Fenwick tree)是一种用于高效求解数列前缀和(Prefix Sum)问题的数据结构,而求逆序对也是一种经典的问题。 假设有个长度为 n 的数组 a,我们要求其中逆序对数量,即所有 i < j 且 a[i] > a[j] 的 (i,j) 对数。这个问题可以用归并排序的思想求解,但是这里我们介绍一种使用树状数组的解法。 思路如下: 1. 将原数组 a 复制一份并排序,得到新数组 b,将 b 中的每个元素在原数组 a 中的下标记录在数组 c 中。 2. 从右往左遍历 a,对于每个元素 a[i],在树状数组中查询前缀和 sum(c[i]-1),即小于 a[i] 的元素个数,将结果累加到逆序对计数器 ans 中。 3. 将 a[i] 在数组 c 中对应的位置在树状数组中更新为 1。 下面是使用 Python 实现的代码: ```python def lowbit(x): return x & -x def update(bit, x, val): n = len(bit) while x <= n: bit[x] += val x += lowbit(x) def query(bit, x): res = 0 while x > 0: res += bit[x] x -= lowbit(x) return res def count_inversions(a): n = len(a) b = sorted(a) c = {v: i+1 for i, v in enumerate(b)} bit = [0] * (n+1) ans = 0 for i in range(n-1, -1, -1): ans += query(bit, c[a[i]]-1) update(bit, c[a[i]], 1) return ans ``` 其中,lowbit 函数是计算 x 的最低位 1 所代表的值;update 函数是树状数组的更新操作;query 函数是树状数组的查询操作;count_inversions 函数是主函数,用于计算逆序对数量。 时间复杂度为 O(nlogn)。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值