给你一个整数数组 nums
和一个整数 k
,请你统计并返回 该数组中和为 k
的子数组的个数 。
子数组是数组中元素的连续非空序列。
示例 1:
输入:nums = [1,1,1], k = 2
输出:2
示例 2:
输入:nums = [1,2,3], k = 3
输出:2
题目很简单对吧,理解题目意思就可以暴力把他做出来。
暴力做法:两层for循环
class Solution { public: int subarraySum(vector<int>& nums, int k) { int res=0; for(int i=0;i<nums.size();i++) { int temp=0; for(int j=i;j<nums.size();j++) { temp+=nums[j]; if(temp==k){ res++; } } } return res; } };
哈希表做法:首先计算一个前缀和,维护一个哈希表(unordered_map)来记录前缀和及其出现的次数。通过在迭代过程中不断更新前缀和,然后查找是否存在符合条件的前缀和(pre - k),来累加符合条件的子数组个数。
class Solution { public: int subarraySum(vector<int>& nums, int k) { int res = 0; // 用于存储符合条件的子数组个数 unordered_map<int, int> mp; // 用于存储前缀和及其出现的次数 int pre = 0; // 当前的前缀和 for (int i = 0; i < nums.size(); i++) { mp[pre]++; // 将当前前缀和的次数加一 pre += nums[i]; // 更新前缀和 // 如果存在前缀和 pre - k,说明从该位置到当前位置的子数组和为 k res += mp[pre - k]; } return res; // 返回符合条件的子数组个数 } };