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

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

链接

问题描述

• 题目：

给你一个正整数数组 nums，请你移除 最短 子数组（可以为空），使得剩余元素的和能被 p 整除。 不允许将整个数组都移除。
请你返回你需要移除的最短子数组的长度，如果无法满足题目要求，返回 -1 。
子数组 定义为原数组中连续的一组元素
来源：力扣（LeetCode）

• 要求:

1 <= nums.length <= $10^5$
1 <= nums[i] <= $10^9$
1 <= p <= $10^9$

思路

1. 本题和974.和可被 K 整除的子数组思路相同
2. 首先求出所有元素和对P的余数Q，也就是说我们只需要在子数组中找到一个和对P的余数为Q，并且要求该子数组的长度最小。
3. 如果该子数组不存在（不能为原数组），返回-1。

代码

//思路：首先计算所有的和，假设余q，找出连续子数组中的最短的除p余q的子数组.(可以进行取余计算避免溢出)
    //对于每一位用一个hash表记录他的位置
    //对于位置index，如果他区域p得到r，首先检查r是否能作为分割的元素，如果能，记录范围，然后继续有两种情况：
    // 1. r在hash表中存在，覆盖，因为之后都是在index之后，如果有对应的答案，那么也是选取index而不是index覆盖掉的元素
    // 2. r不在hash表中，那么直接添加
    //如何判断两个元素x,y相减取余p，得到q呢？:(x-y+p)%p==q
    int[] nums;
    int restNum;
    int p;
    public int minSubarray(int[] nums, int p) {
        this.nums = nums;
        this.p = p;
        HashMap<Integer,Integer> map = new HashMap<>();
        nums[0] %= p;
        int len = nums.length;
        for (int i = 1; i < len; i++) {
            nums[i] = (nums[i]+nums[i-1])%p;
        }
        this.restNum = nums[len-1];
        int res = len;
        if (restNum==0){//说明恰好全部整除
            return 0;
        }
        //记录每一个选择，后选择可以覆盖前面的选择
        for (int i = 0; i < len; i++) {
            int needNum = nums[i]-restNum>=0?nums[i]-restNum:nums[i]-restNum+p;//x-y = restNum||x-y+p = restNum
            if (needNum==0){
                res = Math.min(res,i+1);
            }
            if (map.containsKey(needNum)){
                //如果存在需要的数字，那么就进行计算
                Integer index = map.get(needNum);
                res = Math.min(res,i-index);
            }
            map.put(nums[i],i);
        }
        return res==len?-1:res;
    }