在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007
这道题一开始直接暴力,用插入排序,结果当然是超时了。。。
class Solution {
public:
int InversePairs(vector<int> data) {
int num=0;
for(int i=0;i<data.size()-1;i++){
int j=i+1;
int tmp=data[j];
for(;j>0&&tmp<data[j-1];j--){
data[j]=data[j-1];
if(tmp<data[j-1])
num++;
}
data[j]=tmp;
}
return num;
}
};
后来参考剑指offer,用归并排序,debug了好久。。。
//这道题debug了好久才提交,总是出现各种问题,得到了很多教训,如注释中的1,2,3
//另外debug时,不仅要观察最后的结果,还要看数组是否真的排序了
class Solution {
public:
int InversePairs(vector<int> data) {
//用long long类型否则会超限
long long num = 0;
mergesort(data, num, data.begin(), data.end() - 1);
return num%1000000007;
}
//归并排序
void mergesort(vector<int> &data, long long &num, vector<int>::iterator start, vector<int>::iterator end) {
if (start == end) return;
vector<int>::iterator mid = start + (end - start) / 2;
mergesort(data, num, start, mid);
mergesort(data, num, mid + 1, end);
merge(data, num, start, end);
}
//合并
void merge(vector<int> &data, long long &num, vector<int>::iterator start, vector<int>::iterator end) {
if (start == end) return;
vector<int>::iterator mid = start + (end - start) / 2;
//把要合并的两个子数组保存出来
vector<int> data1(start, mid+1);
vector<int> data2(mid + 1, end+1);
//两个迭代器分别指向两个数组的末尾
vector<int>::iterator p = data1.end() - 1;
vector<int>::iterator q = data2.end() - 1;
vector<int>::iterator m = end;
int sz1 = data1.size();
int sz2 = data2.size();
//合并,当前面数组的元素大于后面时计数加一
while (sz1>0&&sz2>0) {
if (*q<*p) {
num = q - data2.begin()+ 1+num;//1、迭代器相减注意顺序
*m = *p;
--sz1;
--m;
if(sz1>0) --p;//2、有效迭代器的范围是[begin(),end()),begin()位置已经不可以再自减了
}
else {
*m = *q;
--sz2;
--m;
if(sz2>0) --q;
}
}
//3、把剩余的放入数组中,一开始分析出来其中一个数组最多只剩一个元素,而没有让迭代器自减
while (sz1--) {
*m = *p;
if (sz1 > 0) {
--m;
--p;
}
}
while (sz2--) {
*m = *q;
if (sz2 > 0) {
--m;
--q;
}
}
}
};