LeetCode 560.和为k的子数组

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/subarray-sum-equals-k
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

	利用前缀和的思想：有一个数组 int[] arr={2 , 4 , 6 , 1 , 8 , 9 }; 它的前缀和数组为int[]  
	preArr={0 , 2 , 6 , 12 , 13 , 21 , 30 };就是不停的累加，当前数值是前面数的累加；如果要求arr数组某个区间的和，就可以
	sum=preArr[right]-preArr[left]求得；
	回到题目，题目要求求出有多少个子数组的相加和为k,那么只需要遍历前缀和数组，逐个求出sum值，比较统计个数

代码

package test;
import java.util.HashMap;
public class Demo116 {
	public static void main(String[] args) {
		int[] nums= {0,0,0};
		System.out.println(subarraySum(nums, 2));
	}
	
	public static int subarraySum(int[] nums, int k) {
		int len=nums.length;
		int count=0;
		//	前缀和数组
		int[] pre=new int[len+1];
		for(int i=1;i<pre.length;i++) {
			pre[i]=pre[i-1]+nums[i-1];
		}
		
		//遍历统计sum==k个数
		for(int left=0;left<pre.length;left++) {
			for(int right=left;right<pre.length;right++) {
				if(pre[right]-pre[left]==k) {
					count++;
				}
			}
		}
		return count;
    }
}

进阶，用哈希表保存，降低时间复杂度

package test;

import java.util.HashMap;

public class Demo116 {
	public static void main(String[] args) {
		int[] nums= {0,0,0};
		System.out.println(subarraySum(nums, 2));
	}
	
	public static int subarraySum(int[] nums, int k) {
		int len=nums.length;
		int count=0;
		//	前缀和数组
		int[] pre=new int[len+1];
		for(int i=1;i<pre.length;i++) {
			pre[i]=pre[i-1]+nums[i-1];
		}
		
		//类似于 1+2=3  ，3-2=1，所以当我们要求有没有数能够相加等于3，
		//只需要找有没有数是等于1的个数
		HashMap<Integer,Integer> map=new HashMap<>();
 		for(int i=0;i<pre.length;i++) {
			if(map.containsKey(pre[i]-k)) {
				count+=map.get(pre[i]-k);
			}
			map.put(pre[i],map.getOrDefault(pre[i],0)+1);
		}
		
		return count;
    }
}

标签：前缀和 数组 哈希表