Leetcode 3728. Stable Subarrays With Equal Boundary and Interior Sum

• Leetcode 3728. Stable Subarrays With Equal Boundary and Interior Sum
  • 1. 解题思路
  • 2. 代码实现

• 题目链接：3728. Stable Subarrays With Equal Boundary and Interior Sum

1. 解题思路

这一题就是一个前序数组的思路。

我们考察第$j$个元素作为右边界时的情况，则其可用的左边界$i$必然满足以下条件：

• $j-i>=2$
• $c_i=c_j$
• $\sum _{k=0}^j{c}_k-\sum _{k=0}^i{c}_k=2\ast c_j$

因此，我们以组合数$(c_i,\sum _{k=0}^i{c}_k)$来进行考察，那么，对于任意位置$j$作为右边界时其可行的构造方式就是在到$j-2$位置为止，组合数$(c_j,\sum _{k=0}^j{c}_k-2\ast c_j)$出现的次数。

2. 代码实现

由此，我们即可给出python代码实现如下：

class Solution:def countStableSubarrays(self, capacity: List[int]) -> int:n = len(capacity)cumsum = list(accumulate(capacity))cnt = defaultdict(int)ans = 0for i in range(2, n):cnt[(capacity[i-2], cumsum[i-2])] += 1ans += cnt[(capacity[i], cumsum[i]-2*capacity[i])]return ans

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