Leetcode1090. 受标签影响的最大值

最新推荐文章于 2024-05-09 20:43:47 发布

梦想闹钟

最新推荐文章于 2024-05-09 20:43:47 发布

阅读量304

点赞数

分类专栏： Leetcode中等题文章标签： python 开发语言

本文链接：https://blog.csdn.net/qq_43199509/article/details/132619579

版权

Leetcode中等题专栏收录该内容

43 篇文章 0 订阅

订阅专栏

在这里插入图片描述

思路：根据值从大到小排序，然后在加的时候判断是否达到标签上限即可，一开始想用字典做，但是题目说是集合却连续出现两个8，因此使用元组+SortedList进行解决

class Solution:
    def largestValsFromLabels(self, values: list[int], labels: list[int], numWanted: int, useLimit: int) -> int:
        from sortedcontainers import SortedList
        from collections import defaultdict
        sorted_values = SortedList()
        for num in range(0, len(labels)):
            sorted_values.add((values[num], labels[num]))
        total_sum = 0
        label_dict = defaultdict(int)
        for value_label in reversed(sorted_values):
            value = value_label[0]
            label = value_label[1]
            if label_dict[label] < useLimit:
                if numWanted > 0:
                    numWanted -= 1
                    total_sum += value
                    label_dict[label] += 1
                else:
                    return total_sum
        return total_sum