给定两个长度相等的数组 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]
思路:要使其相对于 nums2
的优势最大化,就让nums1[i]能大于nums2[i]时nums1[i]能取到尽量小的值,所以想到对nums1,nums2排序,因为最后要返回一个与nums2原始状态数组算“优势值”,那么要引入一个数组记入nums1下标进行排序、比较
class Solution {
public:
vector<int> advantageCount(vector<int>& nums1, vector<int>& nums2) {
vector<int> num1(nums1.size(),0);
vector<int> num2(nums1.size(),0);
for(int i=0;i<nums1.size();i++)
{
num1[i]=i;
num2[i]=i;
}
sort(num1.begin(),num1.end(),[&](int a,int b)->bool{return nums1[a]<nums1[b];});
sort(num2.begin(),num2.end(),[&](int a,int b)->bool{return nums2[a]<nums2[b];});
vector<int> ret(nums1.size(),INT_MIN);
//保存未匹配上的值
vector<int> lower;
int i=0,j=0;
while(i<nums1.size())
{
while(i<nums1.size()&&nums1[num1[i]]>nums2[num2[j]]){
ret[num2[j]]=nums1[num1[i]];
i++;
j++;
}
if(i<nums1.size()&&nums1[num1[i]]<=nums2[num2[j]])
{
lower.push_back(nums1[num1[i]]);
i++;
}
}
int temp=0;
for(auto& e:ret)
{
e=e!=INT_MIN?e:lower[temp++];
}
return ret;
}
};