Description:
题目大意:寻找数组中的逆序对
解题思路:
算法标签:归并排序
- 利用归并排序的原来
- [3,4] 和 [1,2] 合并的时候,显然 3 大于 1,以及 3 后面的数字也均大于 1,即有 mid - l + 1 个数字
代码:
class Solution {
public:
int count = 0;
void merge(vector<int> &data,int left , int mid , int right) {
// 创建临时数组,使其长度为两个数组加起来的长度
vector<int>temp(right - left + 1);
// 临时数组的下标
int index = 0;
// 原数组起始下标
int begin = left;
// 左,右两边起始下标
int l = left,r = mid + 1;
while(l <= mid && r <= right) {
// 当左边数组的当前元素小的时候,无逆序对
if(data[l] <= data[r]) {
// 存入临时数组
temp[index] = data[l];
index++;
l++;
}
else {
// 存入临时数组
temp[index] = data[r];
// 统计count
count += mid + 1 - l;
count %= 1000000007;
index++;
r++;
}
}
// 左边数组还有元素
while(l <= mid)
temp[index++] = data[l++];
// 右边数组还有元素
while(r <= right)
temp[index++] = data[r++];
// 将临时数组中的元素放入原数组中
for(int i = 0;i < temp.size();i++) {
data[begin++] = temp[i];
}
}
void mergeSort(vector<int> &data , int left , int right) {
int mid = (left + right) >> 1;
if(left < right) {
mergeSort(data, left, mid);
mergeSort(data, mid + 1, right);
merge(data , left , mid , right);
}
}
int InversePairs(vector<int> data) {
if(data.size() < 2)
return 0;
// 进入归并
mergeSort(data , 0 , data.size() - 1);
return count;
}
};