LeetCode 1590. 使数组和能被 P 整除

题目链接：LeetCode 1590. 使数组和能被 P 整除

题意：

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

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

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

 

解题思路：

   用到取余的思想

 

假设 nums 的和除以 P，余数是 mod，

如果 mod == 0，答案就是 0。

如果 mod != 0，答案变成了找原数组中的最短连续子数组，使得其数字和除以 P，余数也是 mod。

由于是求解连续子数组和的问题，很容易想到使用前缀和。

我们可以扫描一遍整个数组，计算到每个元素的前缀和。

假设当前前缀和除以 P 的余数是 curmod，为了找到一段连续子数组对 P 的余数是 mod，我们需要找到一段前缀和，对 P 的余数是 targetmod。其中 targetmod 的求法是：

如果 curmod >= mod，很简单：targetmod = curmod - mod；

如果 curmod < mod，我们需要加上一个 P：targetmod = curmod - mod + P；

const int maxn = 1e5+5;
class Solution {
private:
    long long pre[maxn];
public:
    int minSubarray(vector<int>& nums, int p) {
        int n = nums.size();
        for(int i = 1; i <= n; i++) {
            pre[i] = (pre[i-1] + nums[i-1]) % p ;
        }
        int sum = pre[n-1] % p;
        if(sum == 0) {
            return 0;
        }
        map<int, int> m;
        int ans = 1e9+7;
        for(int i = 1; i <= n; i++) {
            int cur = pre[i-1];
            int target = (cur-sum+p) % p;
            if(m[target]) {
                ans = min(ans, i - m[target]);
            }
            m[cur] = i;
        }
        if(ans == n || ans == 1e9+7) {
            return -1;
        }
        return ans;
    }
};