LeetCode每日一题 | 优势洗牌

Marlowee

于 2022-10-08 16:41:55 发布

阅读量682

点赞数 1

分类专栏： LeetCode 文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/weixin_43427721/article/details/127211054

版权

LeetCode 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

题目描述

给定两个大小相等的数组 $n u m s 1$ 和 $n u m s 2$ ， $n u m s 1$ 相对于 $n u m s 2$ 的优势可以用满足 $n u m s 1 [i]$ > $n u m s 2 [i]$ 的索引 $i$ 的数目来描述。

返回 $n u m s 1$ 的任意排列，使其相对于 $n u m s 2$ 的优势最大化。

样例输入

示例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]

田忌赛马式解题思路：

今以君之下驷与彼上驷，取君上驷与彼中驷，取君中驷与彼下驷。”既驰三辈毕，而田忌一不胜而再胜，卒得王千金。

田忌赛马的故事大家都听说过，简而言之就是发挥每匹优势马的最大效力，并且使用劣势马去消耗对方优势马。

使用相同的策略，我们可以把 $n u m s 1$ 作为田忌的马群，把 $n u m s 2$ 作为齐威王的马群，并遵循以下原则作为 $n u m s 1$ 的上场顺序准则：

首先将两组马群按照脚力小大升序排列，每次派出顺序队列中的一匹马上场；
如果 $n u m s 1$ 中的马匹 $x$ 脚力比得过 $n u m s$ 中的马匹 $y$ ，那么就确定使用当前马匹 $x$ 作为1的优势马匹从前顺序排列，即 $an s [l e f t] = x$ ；
如果 $n u m s 1$ 中的马匹 $x$ 脚力比不过 $n u m s$ 中的马匹 $y$ ，那么就确定使用当前马匹 $x$ 作为1的劣势马匹从后顺序，即 $an s [r i g h t] = x$ ；

如此一来， $n u m s 1$ 中的每匹马都可以根据自身脚力发挥最大效力，并且这是一个可用程序迭代的过程。
唯一需要说明的一点： $n u m s 2$ 中的马匹出战顺序不可再更改，所以要用额外的索引数组对其顺序进行标记，使每次与 $n u m s 1$ 比较的马匹都是顺序递增的。

代码

from typing import List

class Solution:
    def advantageCount(self, nums1: List[int], nums2: List[int]) -> List[int]:

        ans=[0 for i in range(len(nums1))]
        # 比较取巧的idx数组，解决了nums2顺序索引的问题
        idx = sorted(range(len(nums1)),key=lambda i:nums2[i])

        nums1.sort()
        left,right=0,len(nums2)-1

        for i,num in enumerate(nums1):
            # print("第{}次比较过程：".format(i))
            # 优势马对应的情况
            if num>nums2[idx[left]]:
                # print("当前使用nums1中的{}与nums2中索引为{}的{}比较".format(num,idx[left],nums2[idx[left]]))
                ans[idx[left]]=num
                left+=1
            # 劣势马对应的情况
            else:
                # print("当前使用nums1中的{}与nums2中索引为{}的{}比较".format(num, idx[right], nums2[idx[right]]))
                ans[idx[right]]=num
                right-=1
        return ans
if __name__ == '__main__':
    nums1 = [2, 7, 11, 15]
    nums2 = [1, 10, 4, 11]
    print(Solution().advantageCount(nums1, nums2))