Leetcode 3251. Find the Count of Monotonic Pairs II

• Leetcode 3251. Find the Count of Monotonic Pairs II
  • 1. 解题思路
  • 2. 代码实现

• 题目链接：3251. Find the Count of Monotonic Pairs II

1. 解题思路

这一题的话就是一个动态规划的题目，我们给出一个$n\times m$的矩阵cnt，其任意元素cnt[i][j]表示当arr1[i]取值为j时，其可能的取法数目。

显然，对于cnt[n-1]，只要有j < nums[n-1]，都有cnt[n-1][j] = 1，而其他数目为0。

此时，我们只需要给出cnt[i]与cnt[i+1]的递推关系即可。

显然，对于第i个元素，如果我们取值arr1的值为j，则可取的i+1位置上的元素x必须满足：

1. x >= j
2. nums[i+1] - x <= nums[i] - j

即：x >= max(j, nums[i+1]-nums[i]+j)

此时，我们用一个累积数组即可完成上述公示的快速求解。

2. 代码实现

给出python代码实现如下：

MOD = 10**9+7

class Solution:
    def countOfPairs(self, nums: List[int]) -> int:
        n = len(nums)
        m = max(nums)
        cnt = [1 if i <= nums[-1] else 0 for i in range(m+1)]
        
        for i in range(n-2, -1, -1):
            s = list(accumulate(cnt, initial=0))
            cnt = [0 for i in range(m+1)]
            for j in range(nums[i]+1):
                k = max(j, nums[i+1] - nums[i] + j)
                cnt[j] = s[-1] - s[k]
        return sum(cnt) % MOD     

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