知识点:前缀和,哈希表
对于判断子数组的和,可以通过前缀和和哈希表的方式,将每一个前缀和作为key,下标作为value,每次获取到新的前缀和,可以在map中判断是否存在preSum-k的前缀和,也就是二者的区间和为k,若存在,则保存下来该区间的子数组,加入到TreeSet中。此处使用的是TreeSet,目的是按字典序排列,且去除相同子数组。在TreeSet中指定比较器,二者首元素不同时,比较大小,首元素相同时,比较二者的长度,以筛选出不同的子数组。
Java题解如下
import java.util.*;
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param nums int整型一维数组
* @param k int整型
* @return int整型二维数组
*/
public int[][] subarraySum (int[] nums, int k) {
// write code here
int n = nums.length;
Map<Integer, Integer> map = new HashMap<>();
int preSum = 0;
map.put(0, -1);
TreeSet<int[]> set = new TreeSet<>((o1, o2) ->
{
if(o1[0] != o2[0]) {
return o1[0] - o2[0];
}else {
return o1.length - o2.length;
}
});
for(int i = 0; i < n; i++) {
preSum += nums[i];
if(map.containsKey(preSum - k)) {
int start = map.get(preSum - k) + 1;
int[] arr = new int[i - start + 1];
for(int j = start; j <= i; j++) {
arr[j - start] = nums[j];
}
set.add(arr);
}
map.put(preSum, i);
}
int[][] res = new int[set.size()][];
set.toArray(res);
return res;
}
}



京公网安备 11010502036488号