315. 计算右侧小于当前元素的个数 （二分查找+归并排序）

315. 计算右侧小于当前元素的个数

给定一个整数数组 nums，按要求返回一个新数组 counts。数组 counts 有该性质： counts[i] 的值是 nums[i] 右侧小于 nums[i] 的元素的数量。

示例:

输入: [5,2,6,1]
输出: [2,1,1,0]
解释:
5 的右侧有 2 个更小的元素 (2 和 1).
2 的右侧仅有 1 个更小的元素 (1).
6 的右侧有 1 个更小的元素 (1).
1 的右侧有 0 个更小的元素.

**
 * Note: The returned array must be malloced, assume caller calls free().
 */

//二分查找，返回target应该插入的index位置
int findindex(int arr[],int start,int end,int target)
{
  int l = start;
  int r = end - 1;
  while(l <= r) {
      int mid = (l + r)/2;
      if(target > arr[mid]) { //目标值比mid元素大
         l = mid + 1;
      }else { //目标值 <= mid元素
          r = mid - 1;
      }
  }
  return l;
}


//不断插入sorted数组，使之成为一个有序数组
void insert(int *sorted,int start,int end,int index,int value)
{
    int i;
    for(i = end;i >= start;i--) {
       if(i == index) {
          sorted[i] = value; //插入
          break; //就是这个break没加！！！！导致部分结果通过，部分失败
       }
       else if(i != 0) {
           sorted[i] = sorted[i-1]; //后移，给要插入的值腾地
       }
    }

}


int* countSmaller(int* nums, int numsSize, int* returnSize){
    *returnSize = numsSize;
   if(numsSize == 0) return NULL;

   int *sorted = malloc(sizeof(int) * (numsSize));
   int *count = malloc(sizeof(int) * (numsSize));
    memset(sorted,0,sizeof(int) * numsSize);
    memset(count,0,sizeof(int) * numsSize);

    
    int cnt = 0;
    for(int end = numsSize - 1;end >= 0;end--) {
        int index = findindex(sorted,0,cnt,nums[end]);
        count[end] = index;
        insert(sorted,0,cnt,index,nums[end]);
        cnt++;
    }
   
   free(sorted);
   return count;
}