题目链接:
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;
}