LeetCode 1590. 使数组和能被 P 整除(前缀和+map)

题意:
给你一个正整数数组 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;
    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值