思路:
将⻬王和⽥忌的⻢按照战⽃⼒排序,然后按照排名⼀⼀对⽐。如果⽥忌的⻢能赢,那就⽐赛,如果赢不了,那就换个垫底的来送⼈头,保存实⼒。
870. 优势洗牌
给定两个大小相等的数组 A 和 B,
A 相对于 B 的优势可以用满足 A[i] > B[i] 的索引 i 的数目来描述。
返回 A 的任意排列,使其相对于 B 的优势最大化。
示例 1:
输入:A = [2,7,11,15], B = [1,10,4,11]
输出:[2,11,7,15]
示例 2:
输入:A = [12,24,8,32], B = [13,25,32,11]
输出:[24,32,8,12]
提示:
1 <= A.length = B.length <= 10000
0 <= A[i] <= 10^9
0 <= B[i] <= 10^9
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/advantage-shuffle
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
int[] advantageCount(int[] nums1, int[] nums2) {
int n = nums1.length;
// 给 nums2 降序排序
PriorityQueue<int[]> maxQueue = new PriorityQueue<>(
(int[] pair1, int[] pair2) -> {
return pair2[1] - pair1[1];
}
);
for (int i = 0; i < n; i++) {
maxQueue.offer(new int[]{i, nums2[i]});
}
// 给 nums1 升序排序
Arrays.sort(nums1);
// nums1[left] 是最⼩值,nums1[right] 是最⼤值
int left = 0, right = n - 1;
int[] res = new int[n];
while (!maxQueue.isEmpty()) {
int[] pair = maxQueue.poll();
// maxval 是 nums2 中的最⼤值,i 是对应索引
int i = pair[0], maxval = pair[1];
if (maxval < nums1[right]) {
// 如果 nums1[right] 能胜过 maxval,那就⾃⼰上
res[i] = nums1[right];
right--;
} else {
// 否则⽤最⼩值混⼀下,养精蓄锐
res[i] = nums1[left];
left++;
}
}
return res;
}
}
将一个数组降序排序
// 给 nums 降序排序
PriorityQueue<int[]> maxQueue = new PriorityQueue<>(
(int[] pair1, int[] pair2) -> {
return pair2[1] - pair1[1];
}
);
for (int i = 0; i < n; i++) {
maxQueue.offer(new int[]{i, nums[i]});
}