Leetcode 2875. Minimum Size Subarray in Infinite Array

  • Leetcode 2875. Minimum Size Subarray in Infinite Array
    • 1. 解题思路
    • 2. 代码实现
  • 题目链接:Leetcode 2875. Minimum Size Subarray in Infinite Array

1. 解题思路

这一题的话首先由于数组可以无限延伸,因此如果target大于array的总和,那么一定是通过多次循环获得的,因此我们可以提前算好loop数,然后将target变成一组循环里面可能获得的数。

然后,我们就是看怎么最快的得到这个数,这个我们可以通过累积数组来完成,要获得这个target数,只可能是两种情况:

  1. 一组循环内,存在后续某个index上的数使得两者之间的数组之和为target;
  2. 直接取到末尾然后不够的数字从头上进行补充。

我们只需要分别考虑这两种情况,然后将其中可能的实现的方式对应的数组长度进行一下计算然后取最短长度即可。

2. 代码实现

给出python代码实现如下:

class Solution:
    def minSizeSubarray(self, nums: List[int], target: int) -> int:
        n = len(nums)
        s = sum(nums)
        loop = (target // s) * n
        target = target % s
        
        accums = [0] + list(accumulate(nums))
        index = {x: i for i, x in enumerate(accums)}

        res = n+1
        for i, x in enumerate(accums):
            if x + target in index:
                res = min(res, index[x + target] - i)
            if target-(s-x) in index:
                res = min(res, index[target-(s-x)] + n-i)
        return res + loop if res != n+1 else -1

提交代码评测得到:耗时323ms,占用内存36.4MB。

你可能感兴趣的:(leetcode笔记,leetcode,leetcode,2875,周赛,365,循环数组,leetcode,medium)