【
给定两个大小相等的数组 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;
}