力扣 每日一题 870. 优势洗牌难度：中等（贪心+双指针）

Posted 2022-10-10 nefu-ljw

题目链接

https://leetcode.cn/problems/advantage-shuffle/

思路

贪心、双指针。

贪心策略类似于田忌赛马（大的匹配大的，匹配不上就用小的匹配大的）。

首先，肯定要把nums1, nums2排序，对于nums2还要额外记一下排序前的原坐标（可以用pair存到vector里面），存入s2。

然后，以下图为例，用双指针i, j比较两个数组尾部的大小，逆序遍历s2，如果nums1>s2，那么直接匹配上；反之，则将s2匹配上nums1的首部，这个首部的坐标可以再加一个指针k来维护。

代码

class Solution 
public:
    vector<int> advantageCount(vector<int>& nums1, vector<int>& nums2) 
        sort(nums1.begin(),nums1.end());

        vector<pair<int,int>> s2;
        int n=nums1.size();
        for(int i=0;i<n;i++)
            s2.push_back(make_pair(nums2[i],i));
        
        sort(s2.begin(),s2.end()); // 先根据pair中的first排序，若fisrt相同再按second排序

        vector<int> ans(n);
        int i=n-1; // i是nums1的尾部下标
        int k=0; // k是nums1的首部下标
        for(int j=n-1;j>=0;j--) // 逆序遍历s2，j是s2的下标
            int pos=s2[j].second;
            int val=s2[j].first;
            if(nums1[i]>val) // 若当前nums1的尾部比当前s2大
                ans[pos]=nums1[i]; // s2匹配nums1的尾部
                i--;
            else
                ans[pos]=nums1[k]; // s2匹配nums1的首部
                k++;
            
        
        assert(k==i+1);
        return ans;
    
;

/*
[0,1,2,2,4]
[0,0,1,2,3]
ans: [1,2,2,0,4]
*/