# 870. 优势洗牌

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

# 题解

class Solution {
public:
vector<int> advantageCount(vector<int> &nums1, vector<int> &nums2) {
int n = nums1.size(), ids[n];
vector<int> ans(n);
sort(nums1.begin(), nums1.end());
iota(ids, ids + n, 0);
sort(ids, ids + n, [&](int i, int j) { return nums2[i] < nums2[j]; });
int left = 0, right = n - 1;
for (int x : nums1)
ans[x > nums2[ids[left]] ? ids[left++] : ids[right--]] = x;
return ans;
}
};

复杂度分析:

  • 时间复杂度:$$O (nlogn)$$
  • 空间复杂度:$$O (n)$$