在數組中的兩個數字,如果前面一個數字大於後面的數字,則這兩個數字組成一個逆序對。輸入一個數組,求出這個數組中的逆序對的總數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;
}
}
}
};