演算法之陣列中的逆序對
題目:在陣列中的兩個數字如果前面一個數字大於後面的數字,則這兩個數字組成一個逆序對。輸入一個數組,求出這個陣列中的逆序對的總數
例如在陣列{7,5,6,4}中,一共存在5對逆序對,分別是{7,6},{7,5},{7,4},{6,4},{5,4}。
看 到這個題目,我們的第一反應就是順序掃描整個陣列。每掃描到一個數組的時候,逐個比較該數字和它後面的數字的大小。如果後面的數字比它小,則這兩個數字就組成一個逆序對。假設陣列中含有n個數字。由於每個數字都要和O(n)個數字做比較,因此這個演算法的時間複雜度為O(n2)。我們嘗試找找更快的演算法。
//
// 我們以陣列{7,5,6,4}為例來分析統計逆序對的過程,每次掃描到一個數字的時候,我們不能拿它和後面的每一個數字做比較,否則時間複雜度就是O(n2)因此我們可以考慮先比較兩個相鄰的數字。
//
// 如下圖所示,我們先把陣列分解稱兩個長度為2的子陣列,再把這兩個子陣列分別茶城兩個長度為1的子陣列。接下來一邊合併相鄰的子陣列,一邊統計逆序對的數目。在第一對長度為1的子陣列{7},{5}中7大於5,因此{7,5}組成一個逆序對。同樣在第二對長度為1的子陣列{6},{4}中也有逆序對{6,4}。由於我們已經統計了這兩隊子陣列內部逆序對,因此需要把這兩對子陣列排序,以免在以後的統計過程中再重複統計。
//接下來我們統計兩個長度為2的子陣列之間的逆序對。
//
// 我們先用兩個指標分別指向兩個子陣列的末尾,並每次比較兩個指標指向的數字。如果第一個子陣列中的數字大於第二個子陣列中的數字,則構成逆序對,並且逆序對的數目等於第二個子陣列中的剩餘數字的個數。如果第一個陣列中的數字小於或等於第二個陣列中的數字,則不構成逆序對。每一次比較的時候,我們都把較大的數字從後往前複製到一個輔助陣列中去,確保輔助陣列中的數字是遞增排序的。在把較大的數字複製到陣列之後,把對應的指標向前移動一位,接著來進行下一輪的比較。
//
// 經過前面詳細的討論,我們可以總結出統計逆序對的過程:先把陣列分隔成子陣列,先統計出子陣列內部的逆序對的數目,然後再統計出兩個相鄰子陣列之間的逆序對的數目。在統計逆序對的過程中,還需要對陣列進行排序。如果對排序演算法很熟悉,我們不難發現這個排序的過程就是歸併排序。
public class InversePair {
static int count;
public static void main(String[] arr) {
int[] a = {};
int[] p = new int[a.length];
mergesort(a, 0, a.length - 1, p);
}
static void merge(int[] array, int low, int mid, int high, int[] temp) {
int i = low;
int m=mid;
int n=high;
int j = mid + 1;
int index = 0;
while (i <= m && j <= n) {
//左邊的陣列比右陣列大
if (array[i] > array[j]) {
temp[index++] = array[j++];
// 因為如果a[i]此時比右陣列的當前元素a[j]大,
// 那麼左陣列中a[i]後面的元素就都比a[j]大
// 【因為陣列此時是有序陣列】
count += mid - i + 1;
if (count > 1000000007)
{
count %= 1000000007;
}
} else {
temp[index++] = array[i++];
}
}
while (i <= m) {
temp[index++] = array[i++];
}
while (j <= n) {
temp[index++] = array[j++];
}
for (i = 0; i < index; i++) {
array[low + i] = temp[i];
}
}
static void mergesort(int[] a, int low, int high, int[] temp) {
if (low < high) {
int mid = (low + high) / 2;
//左邊有序
mergesort(a, low, mid, temp);
//右邊有序
mergesort(a, mid + 1, high, temp);
//再將兩個有序合併
merge(a, low, mid, high, temp);
}
}
}