题意:
给你一个正整数数组 nums,请你移除 最短 子数组(可以为 空),使得剩余元素的 和 能被 p 整除。
不允许 将整个数组都移除。
请你返回你需要移除的最短子数组的长度,如果无法满足题目要求,返回 -1 。
子数组 定义为原数组中连续的一组元素。
数据范围:
1 <= nums.length <= 1e5
1 <= nums[i] <= 1e9
1 <= p <= 1e9
解法:
设数组和模p的结果为x,
如果x=0,那么不用删除.
否则只需要找到一个长度最小的模p为x的子区间即可,
sum[r]-sum[l-1]=x (mod p)
sum[r]-x=sum[l-1] (mod p)
枚举r,开一个map记录所有sum[l-1]的最右位置,找到最近的sum[r]-x的位置,用区间长度更新答案即可.
code:
class Solution {
public:
int minSubarray(vector<int>& a, int x) {
int t=0;
for(auto s:a)t=(t+s)%x;
if(!t)return 0;
map<int,int>mp;
int n=a.size();
int ans=n+1;
mp[0]=-1;
int sum=0;
for(int i=0;i<n;i++){
sum=(sum+a[i])%x;
int tt=(sum-t+x)%x;
if(mp.count(tt)&&i-mp[tt]!=n){
ans=min(ans,i-mp[tt]);
}
mp[sum]=i;
}
if(ans==n+1)ans=-1;
return ans;
}
};