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

news2024/11/27 13:46:19

一、题目

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

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

子数组定义为原数组中连续的一组元素。
在这里插入图片描述
来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/make-sum-divisible-by-p/description/

二、C++解法

我的思路及代码

我的方法时间复杂度太高，导致不能通过测试用例
采用前缀和的思路，然后从头开始用不同的窗口大小对数组进行遍历，窗口内的元素即为被删除的元素，直到最后剩下的数组元素和可以被整除返回当前的窗口大小，否则返回 -1。由于本题目只需要判断是否可以被整除，所以在前缀和中我们可以简化存储的数据，每次可以存储取余后的数据。

class Solution {
public:
    int minSubarray(vector<int>& nums, int p) {
        int size = nums.size();
        int prefixSum[size+1];
        int ans=1;
        prefixSum[0] = 0;
        for(int i=1;i<nums.size()+1;i++){
            prefixSum[i] = (prefixSum[i-1]+nums[i-1])%p;
        }

        if(prefixSum[size]%p==0)
        return 0;

        while(ans<size+1){
            for(int j=ans;j<size+1;j++){
                if((prefixSum[size]-(prefixSum[j]-prefixSum[j-ans]))%p==0)
                return ans;
            }
            ans++;
        }
        return -1;
    }
};