和爲K的子數組（subarray-sum-equals-k）

給定一個整數數組和一個整數 k，你需要找到該數組中和爲 k 的連續的子數組的個數。

示例 1 :

輸入:nums = [1,1,1], k = 2
輸出: 2 , [1,1] 與 [1,1] 爲兩種不同的情況。

說明 :

數組的長度爲 [1, 20,000]。
數組中元素的範圍是 [-1000, 1000] ，且整數 k 的範圍是 [-1e7, 1e7]。

代碼

其實就是一句sum(i,j)=sum(0,j)-sum(0,i)。 (1+2+3+4+5) - (1+2+3) = 4+5。

class Solution {
    public int subarraySum(int[] nums, int k) {
		int[] sumArr = new int[nums.length];
		int sum = 0;
		for (int i = 0; i < nums.length; i++) {
			sum += nums[i];
			sumArr[i] = sum;
		}

		int res = 0;
		for (int i = 0; i < nums.length; i++) {
			if (sumArr[i] == k)res++;
			for (int j = i + 1; j < nums.length; j++) {
				if (sumArr[j] - sumArr[i] == k)
					res++;
			}
		}
		return res;
    }
}

也可以把其中的for循環換，前綴和+哈希表

public class Solution {
    public int subarraySum(int[] nums, int k) {
        int count = 0, pre = 0;
        HashMap < Integer, Integer > mp = new HashMap < > ();
        mp.put(0, 1);
        for (int i = 0; i < nums.length; i++) {
            pre += nums[i];
            if (mp.containsKey(pre - k))
                count += mp.get(pre - k);
            mp.put(pre, mp.getOrDefault(pre, 0) + 1);
        }
        return count;
    }
}

其他

1248.統計「優美子數組」