Leetcode每日打卡第4天

JAVA_C_P

已于 2024-10-26 15:30:59 修改

阅读量326

点赞数 2

CC 4.0 BY-SA版权

文章标签： leetcode 算法职场和发展

于 2024-10-26 15:30:24 首次发布

本文链接：https://blog.youkuaiyun.com/JAVA_C_P/article/details/143254421

题目

执行操作可获得的最大总奖励 II

给你一个整数数组 rewardValues，长度为 n，代表奖励的值。

最初，你的总奖励 x 为 0，所有下标都是 未标记 的。你可以执行以下操作 任意次 ：

从区间 [0, n - 1] 中选择一个 未标记 的下标 i。
如果 rewardValues[i] 大于你当前的总奖励 x，则将 rewardValues[i] 加到 x 上（即 x = x + rewardValues[i]），并标记下标 i。

以整数形式返回执行最优操作能够获得的最大总奖励。

示例 1：

输入：rewardValues = [1,1,3,3]

输出：4

解释：

依次标记下标 0 和 2，总奖励为 4，这是可获得的最大值。

示例 2：

输入：rewardValues = [1,6,4,3,2]

输出：11

解释：

依次标记下标 0、2 和 1。总奖励为 11，这是可获得的最大值。

提示：

1 <= rewardValues.length <= 5 * 104
1 <= rewardValues[i] <= 5 * 104

思路

python3代码

class Solution:
    def maxTotalReward(self, rewardValues: List[int]) -> int:
        rewardValues.sort()
        if len(rewardValues) >= 2 and rewardValues[-2] == rewardValues[-1] - 1:
            return 2 * rewardValues[-1] - 1
        f = 1
        for x in rewardValues:
            f |= (f & ((1 << x) - 1)) << x
        return f.bit_length() - 1