LeetCode知识点总结 - 870

jeeeee_no

于 2024-01-14 18:00:38 发布

阅读量419

点赞数 8

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

本文链接：https://blog.youkuaiyun.com/m0_59773145/article/details/135586172

版权

LeetCode 专栏收录该内容

417 篇文章

订阅专栏

该文介绍了一种方法，通过先对两个整数数组进行排序，然后根据特定规则生成一个最大化优势的排列，属于中等难度的贪婪算法问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

LeetCode 870. Advantage Shuffle

考点	难度
Greedy	Medium

题目

You are given two integer arrays nums1 and nums2 both of the same length. The advantage of nums1 with respect to nums2 is the number of indices i for which nums1[i] > nums2[i].

Return any permutation of nums1 that maximizes its advantage with respect to nums2.

思路

sort A和B。对于B里的每一个数，如果大于A里最大的，选最大的，否则选A里最小的。

答案

class Solution(object):
    def advantageCount(self, A, B):
        n = len(A)
        B = sorted([(num, i) for i, num in enumerate(B)])[::-1]
        A = sorted(A)[::-1]
        ans = [-1]*n
        
        beg, end = 0, n - 1
        
        for num, ind in B:
            if A[beg] > num:
                ans[ind] = A[beg]
                beg += 1
            else:
                ans[ind] = A[end]
                end -= 1
                
        return ans