Leet-Code(870)--优势洗牌--中等

leetcode.870 优势洗牌

题目描述

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

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

---

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/advantage-shuffle
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

示例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 <= 10^5
• nums2.length == nums1.length
• 0 <= nums1[i], nums2[i] <= 10^9

题目解析

经典田忌赛马

• 下等马去和上等马匹配
• 两个数组从小到达排序，nums1数组以此和nums2数组对比，如果比nums2当前数小，就和nums2中最大的数去匹配

代码

class Solution:
    def advantageCount(self, nums1: List[int], nums2: List[int]) -> List[int]:
        n=len(nums1)
        idx1=list(range(n))
        idx2=list(range(n))
        idx1=sorted(idx1,key=lambda x:nums1[x])
        idx2=sorted(idx2,key=lambda x:nums2[x])
        left=0
        right=n-1
        ans=[0]*n
        for i in range(n):
            if nums1[idx1[i]]>nums2[idx2[left]]:
                ans[idx2[left]]=nums1[idx1[i]]
                left+=1
            else:
                ans[idx2[right]]=nums1[idx1[i]]
                right-=1
        return ans