【C语言刷LeetCode】870. 优势洗牌(M)

【

给定两个大小相等的数组 nums1 和 nums2，nums1 相对于 nums2 的优势可以用满足 nums1[i] > nums2[i] 的索引 i 的数目来描述。

返回 nums1 的任意排列，使其相对于 nums2 的优势最大化。

示例 1：

输入：nums1 = [2,7,11,15], nums2 = [1,10,4,11]
输出：[2,11,7,15]
示例 2：

输入：nums1 = [12,24,8,32], nums2 = [13,25,32,11]
输出：[24,32,8,12]
 

提示：

1 <= nums1.length <= 105
nums2.length == nums1.length
0 <= nums1[i], nums2[i] <= 109

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/advantage-shuffle
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

】

这道题有些难度，因为需要记录原始数组索引，所以刚开始准备用结构体数组来完成，后面其实用二维数组也行。

思路就是：先把两个数组升序排序，如果arr1[i][1]不比arr2[i][1]大的话，就把arr1[i][1]去消耗arr2的最大值。

int Cmp(const void *a, const void *b) // 升序排列
{
    int *x = (int *)a;
    int *y = (int *)b;

    return x[1] > y[1];
}

int* advantageCount(int* nums1, int nums1Size, int* nums2, int nums2Size, int* returnSize){
    int n = nums1Size;
    int arr1[n][2]; // 二维数组，其中arr[n][0]存放索引，arr[n][1]存放数值
    int arr2[n][2];
    int i;
    int l, r;
    int *retarr = malloc(sizeof(int) * n);

    for (i = 0; i < n; i++) {
        arr1[i][0] = i;
        arr1[i][1] = nums1[i];
        arr2[i][0] = i;
        arr2[i][1] = nums2[i];
    }

    qsort(arr1, n, sizeof(arr1[0]), Cmp); // 这里的数组居然可以指针传递
    qsort(arr2, n, sizeof(arr2[0]), Cmp); // arr1和arr2都升序排序一下

/*由于arr2[i][1]可能没有匹配上，而是其最大值被消耗了，所以使用双指针l和r。
如果arr1的首个元素可以增加「优势」，就配对l对应的元素并向右移动一个位置；如果无法配对，就配对r对应的元素并向左移动一个位置*/
    l = 0;    // 这里才是思路重点
    r = n - 1;

    for (i = 0; i < n; i++) {
        if (arr1[i][1] > arr2[l][1]) {
            retarr[arr2[l][0]] = arr1[i][1]; // arr2[l][0]得到nums2的原索引
            l++;
        } else {
            retarr[arr2[r][0]] = arr1[i][1];
            r--;  // 消耗arr2的一个最大值
        }
    }

    *returnSize = n;
    return retarr;
}