import java.util.*; public class Solution { /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param nums int整型一维数组 * @param k int整型 * @return int整型二维数组 */ public int[][] subarraySum (int[] nums, int k) { List<int[]> list = new ArrayList<>(); HashSet<String> set = new HashSet<>(); int n = nums.length; for (int l = 0; l < n; l++) { int sum = 0; for (int r = l; r < n; r++) { sum += nums[r]; if (sum == k) { int[] temp = new int[r - l + 1]; System.arraycopy(nums, l, temp, 0, r - l + 1); if (!set.contains(Arrays.toString(temp))) { list.add(temp); set.add(Arrays.toString(temp)); } } } } return list.toArray(new int[0][0]); } /** 方法二:哈希表 */ 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; } }
本题知识点分析:
1.哈希表存取
2.数组遍历
3.前缀和思想
本题解题思路分析:
1.首先,创建一个列表list和一个哈希集set来存储结果和去重。同时,获取数组nums的长度n。
2.当子数组的和sum等于k时,表示找到一个满足条件的子数组。此时,创建一个长度为r - l + 1的临时数组temp,并使用数组拷贝方法System.arraycopy将nums中的对应元素拷贝到temp中。这里需要使用Arrays.toString将temp转换为字符串形式,以便能够被HashSet识别和比较。
3.然后,判断set是否已经包含了temp,如果是,则表示这个子数组已经添加过了,不需要重复添加;如果不是,则将temp添加到list和set中,同时保证set中也添加了对应的字符串形式。
4.最后集合转二维数组返回即可
5.前缀和是第二种解法