给定两个大小相等的数组 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
非常典型的贪心,每次在A中寻找大于B[i]的最小值,若没有,则返回A中的最小值。如果A组最差的比不过A组最差的,就去消耗A组最强的来扩大优势。
关键在于,如何快速在nums1中找到大于nums2[i]的最小值,可以使用二分法来查找。 所以在查找之前,要将nums1先从小到大排序。
还要注意的一点是每次找到了nums1中大于nums2[i]的最小值,都要在nums1中删除这个值,避免重复使用。这里使用了布尔数组来达到删除效果。
步骤:
- 将num1排序,
- 从num2最大值开始比较,在nums1中二分查找找到刚好比nums2的值大的
- 如果找到,答案数组和nums2下标相同的位置填入上确界,
- 如果找不到,那就找个最小值填入。
private int[] list;// 存入nums1
private boolean[] isVisited; //方便删除元素
public int[] advantageCount(int[] nums1, int[] nums2) {// 主函数
Arrays.sort(nums1);
list = new int[nums1.length];
isVisited = new boolean[nums1.length];
for(int i=0;i<nums1.length;i++) list[i] = nums1[i];
for(int i = nums1.length - 1;i >= 0;i--){
int index = binarySearch(nums2[i]);
nums1[i] = list[index];// 找到了nums2[i]的上确界或者最小值
isVisited[index] = true;// 相当于删除这个被用掉的数字
}
return nums1;
}
//二分查找
public int binarySearch(int val){
int left = 0,right = list.length - 1;
while(left <= right){// 闭区间 ,<= 最后left就是val要插入的位置,也就是val的上确界
int mid = (left + right)/2;
if(list[mid] <= val){// 向右边查找
left = mid +1;
}else{
right = mid - 1;
}
}
while(left < list.length && isVisited[left] == true) left++;
if(left >= list.length) left = 0;// 没在nums1剩余元素中找到nums2[i]的上确界 直接填入最小值
while(left < list.length && isVisited[left] == true) left++;// 寻找找到没被使用的最小值
return left;
}