题目描述
给定两个大小相等的数组 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();
vector<int> idx1(n), idx2(n);
iota(idx1.begin(), idx1.end(), 0);
iota(idx2.begin(), idx2.end(), 0);
sort(idx1.begin(), idx1.end(), [&](int i, int j) { return nums1[i] < nums1[j]; });
sort(idx2.begin(), idx2.end(), [&](int i, int j) { return nums2[i] < nums2[j]; });
//这里的两个sort这样理解
//1.idx1与idx2是nums1与nums2的索引值,也可以理解为下标
//2.此时idx中的元素,也就是具体的数组值各是指向nums中的一个指针,这两个sort的作用就是根据nums中的值对idx中的值进行排序
// 也就是按照真实值的大小对其地址进行排序,也就是根据真实值的大小对其指针进行排序。
//3.下面我们取idx中的最底侧的值也就相当于取到了nums中最小的值。
//4.关于这里的[&](int i, int j) { return nums1[i] < nums1[j]; }
// int i, j 代表了idx中的值
// return nums1[i] < nums1[j];代表了按照nums中的值对idx进行排序
vector<int> ans(n);
int left = 0, right = n - 1;
for (int i = 0; i < n; ++i) {
if (nums1[idx1[i]] > nums2[idx2[left]]) {
ans[idx2[left]] = nums1[idx1[i]];
++left;
}
else {
ans[idx2[right]] = nums1[idx1[i]];
--right;
}
}
return ans;
}
};
力扣题解的完善
观察nums1,全程按照idx从左到右取值,也就是从小到大取值,这里可以直接对nums1进行原地排序,然后后面的步骤相同即可。(也就是没什么大的提升)
修改后代码如下
class Solution {
public:
vector<int> advantageCount(vector<int>& nums1, vector<int>& nums2) {
int n = nums1.size();
vector<int> idx1(n), idx2(n);
iota(idx1.begin(), idx1.end(), 0);
iota(idx2.begin(), idx2.end(), 0);
//sort(idx1.begin(), idx1.end(), [&](int i, int j){return nums1[i] < nums1[j];});
sort(nums1.begin(), nums1.end());
sort(idx2.begin(), idx2.end(), [&](int i, int j){return nums2[i] < nums2[j];});
vector<int> ans(n);
int left = 0, right = n - 1;
for(int i = 0; i < n; i++){
if(nums1[i] > nums2[idx2[left]]){
ans[idx2[left]] = nums1[i];
++left;
}
else{
ans[idx2[right]] = nums1[i];
--right;
}
}
return ans;
}
};