当前位置:   article > 正文

小和问题(归并排序解法)_小和问题(sum) 在一个数组中,每一个数左边比当前数小

小和问题(sum) 在一个数组中,每一个数左边比当前数小

在一个数组中,每一个数左边比当前数小的数累加起来,叫做这个数组的小和。求一个数组 的小和。

例子:

[1,3,4,2,5] 1左边比1小的数,没有; 3左边比3小的数,1; 4左边比4小的数,1、3; 2左边比2小的数,1; 5左边比5小的数,1、3、4、2; 所以小和为1+1+3+1+1+3+4+2=16

 

解题思路: 求一个数左边比它小的数的和,等于求这个数组中,当前数 * 当前数右边比它大的数的个数.

利用归并排序解题,代码如下:

  1. public class SmallSum {
  2. public static int smallSum(int [] arr){
  3. if (arr == null && arr.length < 2){ return -1; }
  4. return smallSum(arr,0,arr.length -1);
  5. }
  6. public static int smallSum(int [] arr, int l,int r){
  7. if (l == r) { return 0; }
  8. int mid = l + ((r - l) >> 1);
  9. return smallSum(arr,l,mid) + smallSum(arr,mid + 1 , r) + merge(arr,l,mid,r);
  10. }
  11. public static int merge(int [] arr, int l, int m, int r){
  12. int [] temp = new int [r - l + 1] ;
  13. int i = 0;
  14. int p1 = l;
  15. int p2 = m + 1;
  16. int res = 0;
  17. while(p1 <= m && p2 <= r){
  18. res += arr[p1] < arr[p2] ? (r - p2 + 1) * arr[p1] : 0 ;
  19. temp[i++] = arr[p1] < arr[p2] ? arr[p1++] : arr[p2++];
  20. }
  21. while(p1 <= m){
  22. temp[i++] = arr[p1++];
  23. }
  24. while(p2 <= r){
  25. temp[i++] = arr[p2++];
  26. }
  27. for (i = 0; i < temp.length ; i ++){
  28. arr[l + i] = temp[i];
  29. }
  30. return res;
  31. }
  32. }

 

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/weixin_40725706/article/detail/227447
推荐阅读
相关标签
  

闽ICP备14008679号