leetcode题解-560. Subarray Sum Equals K

题目:

Given an array of integers and an integer k, you need to find the total number of continuous subarrays whose sum equals to k.

Example 1:
Input:nums = [1,1,1], k = 2
Output: 2
Note:
The length of the array is in range [1, 20,000].
The range of numbers in the array is [-1000, 1000] and the range of the integer k is [-1e7, 1e7].

本题是为了求连续数组和为k的个数。本题需要注意的是可能存在负数。首先第一种方法暴力搜索法,两个循环,但是由于负数的存在不能在碰到一个之后就break,代码如下所示:

    public int subarraySum1(int[] nums, int k) {
        int count=0;
        for(int i=0;iint sum=nums[i];
            if(sum==k){
                count++;
            }
            for(int j=i+1;jsum+=nums[j];
                if(sum==k) count++;
            }
        }
        return count;
    }

另外一种思路就是使用map来保存当前数组的求和值,然后每到一个数都将res进行更新。代码入下:

    public int subarraySum(int[] nums, int k) {
        int sum=0, count=0;
        Map map = new HashMap<>();
        map.put(0, 1);
        for(int i=0; isum += nums[i];
            if(map.containsKey(sum-k))
                count += map.get(sum-k);
            map.put(sum, map.getOrDefault(sum, 0)+1);
        }
        return count;
    }

你可能感兴趣的:(leetcode刷题)