JAVA程序设计：使数组和能被 P 整除（LeetCode：1590）

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

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

子数组 定义为原数组中连续的一组元素。

 

示例 1：

输入：nums = [3,1,4,2], p = 6
输出：1
解释：nums 中元素和为 10，不能被 p 整除。我们可以移除子数组 [4] ，剩余元素的和为 6 。
示例 2：

输入：nums = [6,3,5,2], p = 9
输出：2
解释：我们无法移除任何一个元素使得和被 9 整除，最优方案是移除子数组 [5,2] ，剩余元素为 [6,3]，和为 9 。
示例 3：

输入：nums = [1,2,3], p = 3
输出：0
解释：和恰好为 6 ，已经能被 3 整除了。所以我们不需要移除任何元素。
示例  4：

输入：nums = [1,2,3], p = 7
输出：-1
解释：没有任何方案使得移除子数组后剩余元素的和被 7 整除。
示例 5：

输入：nums = [1000000000,1000000000,1000000000], p = 3
输出：0
 

提示：

1 <= nums.length <= 105
1 <= nums[i] <= 109
1 <= p <= 109

思路：比较简单，前缀和是一定要求的，我们假设当前前缀和为x，则我们需要去掉一段子数组并以当前元素结尾需要满足：(x-y)%p=sum%p；其中y为之前的某个前缀和，sum为数组的总和，剩下的就是用map存一下就ok了。

class Solution {
    public int minSubarray(int[] nums, int p) {

        int ans;
        int n = nums.length;
        long sum = 0, now = 0;
        Map<Long, Integer> map = new HashMap<>();

        for (int i = 0; i < n; i++)
            sum += nums[i];

        if (sum % p == 0) return 0;

        ans = n;
        long mod = sum % p;
        map.put(0L, -1);

        for (int i = 0; i < n; i++) {
            now += nums[i];
            if (map.containsKey((now % p - sum % p + p) % p))
                ans = Math.min(ans, i - map.get((now % p - sum % p + p) % p));
            map.put(now % p, i);
        }

        return ans == n ? -1 : ans;

    }
}