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.前缀和是第二种解法

京公网安备 11010502036488号