數組中的逆序對

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


發佈了133 篇原創文章 · 獲贊 23 · 訪問量 5萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章