题目描述
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007
解题思路
1、暴力解法,顺序扫描整个数组,如果碰到比他小的数字则计数加一,很显然这种算法的复杂度为n^2的,不适合。
class Solution {
public:
int InversePairs(vector<int> data) {
int count = 0;
for(int i = 0; i<data.size(); i++){
int current_num = data[i];
for(int j = i+1; j<data.size();j++){
if(data[j] < current_num)
count ++;
}
}
return count%1000000007;
}
};
2、归并排序思路见书P250
class Solution {
public:
int InversePairs(vector<int> data) {
if(data.size() == 0)
return 0;
vector<int> copy;
for(int i = 0; i<data.size(); i++)
copy.push_back(data[i]);
return InversePairs(data, copy, 0, data.size()-1)%1000000007;
}
long InversePairs(vector<int> &data, vector<int> ©, int begin, int end){
if(begin == end){
copy[begin] == data[end];
return 0;
}
int mid = (begin + end) >> 1;
//使左半端有序,并计算逆序对的对数,注意传进去的为排好序的数组
long leftCount = InversePairs(copy, data, begin, mid);
//使右半段有序,并计算逆序对数目
long rightCount = InversePairs(copy, data, mid+1, end);
int i = mid; // i初始化为前半段最后一个数字的下标
int j = end; // j初始化为后半段最后一个数字的下标
int indexcopy = end; // 辅助数组复制的数组的最后一个数字的下标
long count = 0; // 计数,逆序对的个数,注意类型
while(i >= begin && j >= mid+1){
if(data[i] > data[j]){
copy[indexcopy--] = data[i--];
count += j-mid;
}
else{
copy[indexcopy--] = data[j--];
}
}
for(;i >= begin; --i){
copy[indexcopy--] = data[i];
}
for(;j >= mid+1 ; --j){
copy[indexcopy--] = data[j];
}
return leftCount + rightCount + count;
}
};