题目:数组中的逆序对
思路:
典型的归并排序(很好的题目)
这是一个归并排序的合并过程,主要考虑合并两个有序序列时,计算逆序对的个数!
对于两个升序序列,设置两个下标(前下标和后下标),初始化为前序列第一个数字的下标和后序列第一个数字的下标。如果前下标对应的值大于后下标对应的值,则有“前序列剩下的长度”个逆序对(即mid - first_start + 1个),否则这不构成逆序对。然后把较大值拷贝到辅助数组中,即最终把两个有序序列合并到辅助数组,并保证在辅助数组中也是升序的。
#include<iostream>
#include<vector>
using namespace std;
//思路是归并排序的思想。计算左、右部分的逆序对的时候,利用递归。(个人感觉类似DFS,从最底层子递归入手)
class Solution {
public:
int InversePairs(vector<int> data) {
if (data.size() == 0)
return 0;
return merge(data, 0, data.size() - 1);
}
private:
int merge_count(vector<int> &data, int start, int mid, int end)//此时左、右部分都是升序序列
{
vector<int> temp(data);//辅助数组
int count = 0;
int first_start = start;//初始化为前半段第一个数字的下标
int second_start = mid + 1;//初始化为后半段第一个数字的下标
int copy_start = start;//初始化为辅助数组第一个数字的下标
while (first_start <= mid&&second_start <= end)
{
if (data[first_start] <= data[second_start])
temp[copy_start++] = data[first_start++];
else
{
temp[copy_start++] = data[second_start++];
count = count + mid - first_start + 1;
}
}
//若second_start先到了后半段末尾
while (first_start <= mid)
temp[copy_start++] = data[first_start++];
//若first_start先到了前半段末尾
while (second_start <= end)
temp[copy_start++] = data[second_start++];
for (int i = start; i <= end; i++)
data[i] = temp[i];
return count;
}
int merge(vector<int> &data, int start, int end)//左、右部分分别递归
{
if (start == end)//子递归结束条件
return 0;
int mid = (start + end) / 2;
int left = merge(data, start, mid);
int right = merge(data, mid + 1, end);
int count = merge_count(data, start, mid, end);
return left + right + count;
}
};
int main(void) {
vector<int>a = { 7,5,6,4 };
Solution solution;
int result = solution.InversePairs(a);
cout << result << endl;
system("pause");
return 0;
}