小和問題

在一個數組中,每一個元素左邊比當前元素值小的元素值累加起來,叫做這個數組的小和
例如:[2,3,4,1,5]
2左邊比2小的元素:無
3左邊比3小的元素:2
4左邊比4小的元素:2,3
1左邊比1小的元素:無
5左邊比5小的元素:2,3,4,1
小和small_sum = 2 + 2 + 3 + 2 + 3 + 4 + 1 = 17

思路:使用歸併算法的思想進行求解

public static int smallSum(int[] arr){
    if(arr == null || arr.length < 2){
        return 0;    
    }
    return mergrSort(arr, 0, arr.length - 1);
} 
public static int mergeSort(int[] arr, int L, int R){
    if(L == R) return 0;
    int mid = L + (R - L) >> 2; //找到中點
    return mergeSort(arr, L, mid) + mergeSort(arr, mid + 1, R) + merge(arr, L, mid, R);
}
public static merge(int[] arr, int L, int mid, int R){
    int[] help = new int[R - L + 1];
    int p1 = L;
    int p2 = mid + 1;
    int res = 0;
    int i = 0;
    while(p1 <= m && p2 <= R){
        res += arr[p1] < arr[p2] : (r - p2 + L) * arr[p1] : 0;
        //將排序後的數放入輔助數組
        help[i++] = arr[p1] < arr[p2] ? arr[p1++] : arr[p2++];
    }
    while(p1 <= L){//若p2耗盡則將p1中剩餘的數字填入help中
        help[i++] = arr[p1++];
    }
    while(p2 <= R){//若p1耗盡則將p2中剩餘的數字填入help中
        help[i++] = arr[p2++];
    }
    for(int i = 0; i < help.length; i++){//將輔助數組中的數拷貝到原數組中
        arr[L + i] = help[i];
    }
    return res;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章