leetcode【每日一题】560. 和为K的子数组 java

题干

给定一个整数数组和一个整数 k，你需要找到该数组中和为 k 的连续的子数组的个数。

示例 1 :

输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
说明 :

数组的长度为 [1, 20,000]。
数组中元素的范围是 [-1000, 1000] ，且整数 k 的范围是 [-1e7, 1e7]。

想法

思路和算法

考虑以 ii 结尾和为 kk 的连续子数组个数，我们需要统计符合条件的下标 jj 的个数，其中 0\leq j\leq i0≤j≤i 且 [j…i][j…i] 这个子数组的和恰好为 kk 。

我们可以枚举 [0…i][0…i] 里所有的下标 jj 来判断是否符合条件，可能有读者会认为假定我们确定了子数组的开头和结尾，还需要 O(n)O(n) 的时间复杂度遍历子数组来求和，那样复杂度就将达到 O(n^3)O(n
3
) 从而无法通过所有测试用例。但是如果我们知道 [j,i][j,i] 子数组的和，就能 O(1)O(1) 推出 [j-1,i][j−1,i] 的和，因此这部分的遍历求和是不需要的，我们在枚举下标 jj 的时候已经能 O(1)O(1) 求出 [j,i][j,i] 的子数组之和。

思路参考

Java代码

package daily;

public class SubarraySum {
    public int subarraySum(int[] nums, int k) {
        int count = 0;
        for (int start = 0; start < nums.length; ++start) {
            int sum = 0;
            for (int end = start; end >= 0; --end) {
                sum += nums[end];
                if (sum == k) {
                    count++;
                }
            }
        }
        return count;
    }


}