洛谷 逆序对(归并排序|树状数组)

题目链接:

逆序对 - 洛谷

leetcode也有原题,考虑到leetcode题解写的好,也贴个链接:力扣 - 数组中的逆序对

思路:

在归并排序中,除了数组最左边的元素,其他元素都有机会出现在将要合成的两个数组的右半边数组中,在“并”的过程中,需要逐个比较左右两个数组的数字,在这个比较的过程,可以顺便计算逆序数数量。

举个栗子,左半边数组是1 6 8 9,右半边是1 3 4 5,i是左数组指针,j是右数组指针。

当 i 指向6,j 指向3的时候,要把3放进数组对吧,但是在这之前我们加一步,由于数组有序,6比3大,那么在左数组中,从6开始往右的数字都比3大,即如果arr[ i ] > arr[ j ],就说明[ i, mid ]范围内所有arr值都比arr[ j ]大,所以逆序数要加mid-i+1个。

其余代码部分和归并排序代码并无二致。

本题还有树状数组的做法,有空再加上。(update:树状数组解法已添加)

代码:

#include <bits/stdc++.h>
using namespace std;
const int maxn = 5e5+5;
int n, arr[maxn], tmp[maxn];
long long sum = 0; //sum可能很大,记得开long long

void merge(int left, int right){
    if(left >= right) return; //长度为1,退出分治
    int mid = left + (right-left)/2;
    merge(left, mid); merge(mid+1, right);
    int i=left, j=mid+1, t=left;
    //左右两边都有数字,要比较后写入
    //如果遇到左边比右边大,要记录增加的逆序数,此时[i,mid]部分的arr值都比arr[j]大,逆序数加上mid-i+1
    while(i<=mid && j<=right){
        if(arr[i] <= arr[j]) tmp[t++] = arr[i++];
        else {sum += mid-i+1; tmp[t++] = arr[j++];} //如果把这里的sum += mid-i+1去掉,就是纯的归并排序啦
    }
    //处理剩下的数字,此时只剩左边或右边,直接写入即可
    while(i<=mid) tmp[t++] = arr[i++];
    while(j<=right) tmp[t++] = arr[j++];
    for(int k=left; k<=right; k++) arr[k] = tmp[k]; //排序结果录入arr中
}
int main(){
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    cin >> n;
    for(int i=1; i<=n; i++) cin >> arr[i];
    merge(1, n); //从1~n范围内进行归并排序
    cout << sum;
}

2022.2.27 update:更新树状数组做法(离散化+树状数组)

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N=5e5+5;
int n, a[N], t[N];
vector<int> v;

inline int lowbit(int x){return x&(-x);}
void add(int p){while(p<=n){t[p]++; p+=lowbit(p);}}
inline int ask(int p){int res=0; while(p) res+=t[p], p-=lowbit(p); return res;}
signed main(){
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    //1.输入并离散化
    cin>>n; for(int i=1; i<=n; i++) cin>>a[i], v.push_back(a[i]);
    sort(v.begin(), v.end());
    for(int i=1; i<=n; i++)
        a[i] = lower_bound(v.begin(), v.end(), a[i])-v.begin()+1;
    //2.开始计算答案
    int ans = 0;
    for(int i=1; i<=n; i++){
        add(a[i]);
        ans += i-ask(a[i]);
    }
    cout<<ans;
}

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值