题目链接](剑指 Offer II 010. 和为 k 的子数组 - 力扣(LeetCode) (leetcode-cn.com))
思路1
-
先将前i个元素子数组之和存下来
-
再用双重循环sum[j]-sum[i]求下标j和下标i之间的子数组和是否为k ,时间复杂度为
O ( n 2 ) O(n^2) O(n2)
代码1
class Solution {
public int subarraySum(int[] nums, int k) {
int[] sum = new int[nums.length+1];
int count = 0;
for(int i = 0; i < nums.length; i++) {
if(i > 0) sum[i] = nums[i] + sum[i-1];
else sum[0] = nums[0];
if(sum[i] == k) count++;
}//前i个子数组之和
for(int j = nums.length-1; j >= 0; j--) {
for(int i = 0; i < j; i++) {
if(sum[j] - sum[i] == k) count++;
}
}
return count;
}
}
思路2
-
用HashMap将前num个数的子数组和出现的次数存下来
-
假设当前前num个数的和为ans,则看前num-1个数中有多少个子数组的和为ans-k,同时剩下的子数组的和就为k
-
时间复杂度为
O ( N ) O(N) O(N)
代码2
public int subarraySum(int[] nums, int k) {
Map<Integer,Integer> sum = new HashMap<>();
sum.put(0,1);
int ans = 0;
int count = 0;
for(int num : nums) {
ans += num;//前num个数之和
count += sum.getOrDefault(ans-k,0); //前num-1个数中子数组和为ans-k的个数,如果hashmap中不存在ans-k,返回默认值0;
//存在多少个子数组和为ans-k,即存在多少个子数组和为k
sum.put(ans,sum.getOrDefault(ans,0)+1);//将当前前num个数之和放入hashmap
}
return count;
}
Java知识
HashMap
-
HashMap 是一个散列表,它存储的内容是键值对(key-value)映射。
-
实现了 Map 接口。
-
是无序的,即不会记录插入的顺序。
-
HashMap 的 key 与 value 类型可以相同也可以不同,可以是字符串(String)类型的 key 和 value,也可以是整型(Integer)的 key 和字符串(String)类型的 value。[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-SxhX1lo0-1641995638413)(https://static.runoob.com/images/mix/java-map.svg)]
-
HashMap 类位于 java.util 包中,使用前需要引入它,语法格式如下:
import java.util.HashMap; // 引入 HashMap 类
-
创建HashMap对象
Map<Integer,Integer> sum = new HashMap<>(); // or HashMap<Integer, String> Sites = new HashMap<Integer, String>();
添加键值对(key-value) —put() 方法
访问元素—用 get(key) 方法来获取 key 对应的 value
删除元素—用 remove(key) 方法来删除 key 对应的键值对(key-value)
删除所有键值对(key-value)—clear 方法
// 创建 HashMap 对象 Sites
HashMap<Integer, String> Sites = new HashMap<Integer, String>();
// 添加键值对
Sites.put(1, "Google");
Sites.put(2, "Runoob");
Sites.put(3, "Taobao");
Sites.put(4, "Zhihu");//添加元素
System.out.println(Sites.get(3));//访问键为3元素
Sites.remove(4);//删除元素
Sites.clear();//删除所有元素
getOrDefault() 方法
getOrDefault() 方法获取指定 key 对应对 value,如果找不到 key ,则返回设置的默认值。
getOrDefault() 方法的语法为:
hashmap.getOrDefault(Object key, V defaultValue)