题解 | #子群的标签和#
子群的标签和
https://www.nowcoder.com/practice/4058a95b317f4f0e87e5a1bfa6db9aad
知识点:前缀和,哈希表
对于判断子数组的和,可以通过前缀和和哈希表的方式,将每一个前缀和作为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; } }