[LeetCode解题报告] 1590. 使数组和能被 P 整除

一、 题目

1. 题目描述

给你一个正整数数组 nums，请你移除 最短 子数组（可以为 空），使得剩余元素的 和 能被 p 整除。 不允许 将整个数组都移除。

请你返回你需要移除的最短子数组的长度，如果无法满足题目要求，返回 -1 。

子数组 定义为原数组中连续的一组元素。

 

示例 1：

输入：nums = [3,1,4,2], p = 6
输出：1
解释：nums 中元素和为 10，不能被 p 整除。我们可以移除子数组 [4] ，剩余元素的和为 6 。

示例 2：

输入：nums = [6,3,5,2], p = 9
输出：2
解释：我们无法移除任何一个元素使得和被 9 整除，最优方案是移除子数组 [5,2] ，剩余元素为 [6,3]，和为 9 。

示例 3：

输入：nums = [1,2,3], p = 3
输出：0
解释：和恰好为 6 ，已经能被 3 整除了。所以我们不需要移除任何元素。

示例  4：

输入：nums = [1,2,3], p = 7
输出：-1
解释：没有任何方案使得移除子数组后剩余元素的和被 7 整除。

示例 5：

输入：nums = [1000000000,1000000000,1000000000], p = 3
输出：0

 

提示：

• 1 <= nums.length <= 105
• 1 <= nums[i] <= 109
• 1 <= p <= 109

Related Topics

• 数组
• 哈希表
• 前缀和

• 👍 57
• 👎 0

### 2. 原题链接 链接: [1590. 使数组和能被 P 整除](https://leetcode.cn/problems/make-sum-divisible-by-p/)

二、 解题报告

1. 思路分析

看题解才会做：

1. 用哈希表维护每个前缀和模p结果的最近下标
2. 那么每次都可以去查找一下当前模-mod对应的值是否存在，存在表示满足需求可以移除。
3. 复杂度O(n)。

2. 复杂度分析

最坏时间复杂度O(n)

3. 代码实现

前缀和+哈希表。

class Solution:
    def minSubarray(self, nums: List[int], p: int) -> int:
        n = len(nums)
        total = sum(nums)
        mod = total%p
        if mod == 0:
            return 0
        elif n == 1:
            return -1

        # 如果某个子数组的和%p==mod，就可以移除。        
        # 计算前缀和的过程中，用字典记录前边每个和模p结果的最近下标，
        # 如果当前模-mod存在，那么这个差就是这个区间的和，模==mod
        ans = n  # 不允许移除整个数组
        presum = 0
        pres = {0:-1}
        for i in range(n):
            presum += nums[i]
            cur_mod = presum%p
            need_mod = (cur_mod+p-mod)%p
            # print(presum,cur_mod,need_mod)
            need_idx = pres.get(need_mod,-2)
            if need_idx != -2 :
                ans = min(ans,i-need_idx)
                if ans == 1:
                    return 1

            pres[cur_mod] = i           

        if ans == n:  # 不允许移除整个数组
            return -1

        return ans

三、 本题小结

1. 还需要一定数论的同余知识。

四、 参考链接

• 链接: [英雄星球六月集训LeetCode解题日报] 第八日 前缀和