力扣题解(优势洗牌)

给定两个长度相等的数组 nums1 和 nums2nums1 相对于 nums2 的优势可以用满足 nums1[i] > nums2[i] 的索引 i 的数目来描述。

返回 nums1 的任意排列,使其相对于 nums2 的优势最大化。

思路:田忌赛马的变种,从nums1中取一个数,去找nums2中小于其的最大的数,然后匹配对于的位置。若没有比nums1的数小的,则用其去和nums2中最大的数匹配对应的位置,这样就能实现最大程度的优势。但是,对于nums1中每一个数,去nums2中找对应的数需要的时间是o(n),这样总体的时间复杂度就是o(n^2),优化的点在于找nums2中对应的位置。

优化的思路是对nums1进行排序,对nums2进行伪排序(就是用一个数组记录nums2从小到大的对应的下标),这样对应位置要么是小于nums1的数,就从num2的当前位置替换。要么是当前所有数都大于,都大于的时候就从nums2的最大数位置替换。

对于伪排序部分,使用了lambda表达式语法,[&]表示引用捕捉父作用域中的所有变量,这样就能用到nums2,(int i,int j)是参数部分,{内部是函数体},返回类型忽略,由编译器自行推导。

本题的核心难点一是田忌赛马的思路,即匹配谁的问题,一个是对nums2的伪排序方法。

class Solution {
public:
    vector<int> advantageCount(vector<int>& nums1, vector<int>& nums2) {
     int n=nums1.size();
     sort(nums1.begin(),nums1.end());
     vector<int>index(n);
     for(int i=0;i<n;i++)
     {
        index[i]=i;
     }
     sort(index.begin(),index.end(),[&](int i,int j)
     {
        return nums2[i]<nums2[j];
     });

     vector<int>ret(n);
     int r=n-1,l=0;
     for(auto e:nums1)
     {
       if(e>nums2[index[l]])
       {
        ret[index[l++]]=e;
       }
       else
       {
         ret[index[r--]]=e;

       }
     }
     
     return ret;
    }
};

  • 3
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值