题目链接: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;
}
};