贪心之田忌赛马

今天遇到一道算法题870优势洗牌,发现这个问题类似著名的田忌赛马问题。

使用贪心算法可以有效解决这个问题,其思想核心是:比得过就比,比不过就找个最菜的垫背。

下面的C++代码也值得品味,使用了Lambda表达式。

class Solution {
public:
    vector<int> advantageCount(vector<int> &nums1, vector<int> &nums2) {
        int len = nums1.size();
        vector<int> ans(len);    //结果集
        /*记录nums2的数字和下标*/
        vector<pair<int, int>> nums3;
        nums3.reserve(len); //预留vector空间
        for (int i = 0; i < len; i++)
            nums3.emplace_back(nums2[i], i);
        /*对nums1和nums3进行由大到小排序*/
        sort(nums3.begin(),nums3.end(),[](pair<int,int>&a,pair<int,int>&b){
            return a.first>b.first;
        });
        sort(nums1.rbegin(),nums1.rend());  //逆序排列
        /*双指针进行比较*/
        int left=0,right=len-1;
        for(auto &item:nums3){
            if(item.first<nums1[left]){
                ans[item.second]=nums1[left];
                left++;
            }else{
                ans[item.second]=nums1[right];
                right--;
            }
        }
        return ans;

    }
};

注意:这里的vector.reserve(储备),是在已知长度的情况下,为vector预留空间,这样后序添加的时候,就无需频繁扩容(含有数组的赋值),提高效率。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值