小和问题求解
小和問題
- 題目:在一個數組中,每一個數左邊比當前數小的數累加起來,叫做這個數組的小和。
- 舉例:
1 3 4 2 5
小和為:1 + (1 + 3) + 1 + (1 + 3 + 4 + 2) = 16
代碼實現
普通的暴力破解為:N + N-1+ N-2 + N-3 + ···+ 1 時間復雜度為:O(N^2)
借助歸并排序可以實現時間復雜度為:O(N*logN)
歸并排序詳解
package LeetCode;/*** @Description:* @PackageName: LeetCode* @ClassName: smallSum* @Author: Y-peak* @Date: 2021.08.28 00:08 星期六*/public class smallSum {public static int smallSum(int[] arr){if(arr == null || arr.length < 2){return 0;}return process(arr, 0 , arr.length-1);}public static int process(int[] arr, int L,int R){if(L == R){return 0;}int mid = L + ((R-L) >> 1);return process(arr, L, mid)+ process(arr, mid + 1, R)+ merge(arr, L, mid, R);}public static int merge(int[] arr, int L, int M, int R){int[] help = new int[R-L+1];int i = 0;int p1 = L;int p2 = M + 1;int res = 0;while (p1 <= M && p2 <= R){res += arr[p1] < arr[p2] ? (R - p2 + 1) * arr[p1] : 0;help[i++] = arr[p1] <= arr[p2] ? arr[p1++] : arr[p2++];}while (p1 <= M){help[i++] = arr[p1++];}while (p2 <= R){help[i++] = arr[p2++];}for(i = 0; i < help.length; i++){arr[L+i] = help[i];}return res;}public static void main(String[] args) {int[] arr = new int[]{1,3,4,2,5};System.out.println(smallSum(arr));} }總結
- 上一篇: 十大排序算法之归并排序
- 下一篇: 十大排序算法之快速排序(两种方法)