【LeetCode】1090. 受标签影响的最大值

说文科技

已于 2024-05-09 23:56:16 修改

阅读量355

点赞数 1

CC 4.0 BY-SA版权

分类专栏： # LeetCode # python 文章标签： leetcode 算法数据结构

于 2023-05-23 22:45:18 首次发布

喜欢文章？请私信联系作者。

本文链接：https://blog.csdn.net/liu16659/article/details/130836364

LeetCode 同时被 2 个专栏收录

216 篇文章

订阅专栏

python

35 篇文章

订阅专栏

该文章介绍了一个使用贪心策略来解决问题的算法，首先对values和labels数组按值排序，然后通过限制每个标签下的元素数量（不超过useLimit），计算可以得到的最大和。在Python中，这个过程涉及列表排序、字典计数以及循环遍历来满足numWanted个元素的要求。

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

1.题目

题目

2. 方法

贪心法即可。主要思想如下：

step1. 将values数组和labels数组按照顺序排序，排序规则是按照values中的大小。并将排好序的信息放到一个list中，该list中的每项都是一个tuple
step2.按照 numWanted 和 useLimit 计算可以得到的最大和。

3. 代码

具体代码如下，需要注意的是：对list排序的实现使用的key与reverse。

class Solution:
    def largestValsFromLabels(self, values: List[int], labels: List[int], numWanted: int, useLimit: int) -> int:
        # step1.得到一个数组
        arr =[(val,lab) for val,lab in zip(values,labels)] 

        # step2.  然后将数组排序，按照值的大小
        # 注意语法
        arr = sorted(arr,key = lambda x:x[0], reverse=True) 

        # step 3. 对dict进行初始化
        # 这里可以直接使用defaultdict(int) 代替显式初始化
        lab2dic = {}
        for lab in labels:
            lab2dic[lab] = 0

        # step4. 开始计数求和
        total = 0
        cnt = 0
        for i,j in arr:
            if cnt < numWanted: # 如果尚未放满
                if lab2dic[j] < useLimit: 
                    lab2dic[j] += 1  # 对值为j的数进行计数，不能超过useLimit
                    total += i
                    cnt += 1 # 放数+1
            else:
                break    
        return total