Java實現 LeetCode 523 連續的子數組和(ง •_•)ง

523. 連續的子數組和

給定一個包含非負數的數組和一個目標整數 k,編寫一個函數來判斷該數組是否含有連續的子數組,其大小至少爲 2,總和爲 k 的倍數,即總和爲 n*k,其中 n 也是一個整數。java

示例 1:數組

輸入: [23,2,4,6,7], k = 6
輸出: True
解釋: [2,4] 是一個大小爲 2 的子數組,而且和爲 6。
示例 2:ide

輸入: [23,2,6,4,7], k = 6
輸出: True
解釋: [23,2,6,4,7]是大小爲 5 的子數組,而且和爲 42。
說明:函數

數組的長度不會超過10,000。
你能夠認爲全部數字總和在 32 位有符號整數範圍內。測試

PS:
當求是他的倍數的時候根據餘數求
測試用例0 0是個煩人的點spa

class Solution {
       public boolean checkSubarraySum(int[] nums, int k) {
        
        
        if(nums.length < 2) return false;
        for(int i = 0; i < nums.length-1; ++i) 
            if(nums[i] == 0 && nums[i+1] == 0) return true;
        if(k == 0) return false;
        if(k < 0) k = -k;
        
        Map<Integer, Integer> map = new HashMap<>();
        map.put(0, -1);
        int sum = 0;
        for(int i = 0; i < nums.length; ++i) {
            sum += nums[i];
            int mod = sum % k;
            if(map.containsKey(mod)) {
                if(i-map.get(mod) > 1)
                    return true;
            }
            else // 不存在再更新
                map.put(mod, i);
        }
        return false;
    }
}
相關文章
相關標籤/搜索