Leetcode 3041. Maximize Consecutive Elements in an Array After Modification

• Leetcode 3041. Maximize Consecutive Elements in an Array After Modification
  • 1. 解题思路
  • 2. 代码实现

• 题目链接：3041. Maximize Consecutive Elements in an Array After Modification

1. 解题思路

这一题思路上同样就是一个动态规划，我们首先将原数组进行有序排列。

然后，我们只需要考察每一个元素作为序列起始位置时最多可能的取法即可。

需要注意的是：

• 每一个元素作为序列开头的可能性有两种：使用原始的值或者使用该值加一的结果；
• 如果某一个数作为下一个目标值时，如果有某个元素不为当前的元素且为目标值减一，那么优先取这个数，因为目标值还可以有他用，而这个值如果有用只能被用在这里。

此时，我们即刻给出我们的代码实现。

2. 代码实现

给出python代码实现如下：

class Solution:
    def maxSelectedElements(self, nums: List[int]) -> int:
        n = len(nums)
        nums = sorted(nums)

        @lru_cache(None)
        def dp(idx, nxt):
            if idx >= n:
                return 0
            i = bisect.bisect_right(nums, nxt-1)-1
            if i < n and i > idx and nums[i] == nxt-1:
                return 1 + dp(i, nxt+1)
            j = bisect.bisect_left(nums, nxt)
            if j < n and j > idx and nums[j] == nxt:
                return 1 + dp(j, nxt+1)
            return 0
        
        ans0 = 1 + max(dp(idx, nums[idx]+1) for idx in range(n))
        ans1 = 1 + max(dp(idx, nums[idx]+2) for idx in range(n))
        return max(ans0, ans1)

提交代码评测得到：耗时1326ms，占用内存274.5MB。