和为K的子数组

问题描述 :
给定一个整数数组和一个整数 k，你需要找到该数组中和为 k 的连续的子数组的个数。

示例 1 :
输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。

输入说明 :
首先输入nums数组的长度n，
然后输入n个整数，以空格分隔。
最后输入整数k。
n的范围为 [1, 20000]。
数组中元素的范围是 [-1000, 1000] ，且整数 k 的范围是 [-1e7, 1e7]。

输出说明 :
输出一个整数

输入范例 :
3
1 1 1
2

输出范例 :
2

#include<iostream>
#include<vector>
#include<unordered_map>
using namespace std;

int subarraySum(vector<int> &nums,int k){
	unordered_map<int, int> mp;
        mp[0]=1;
        int count =0,pre=0;
        for (int i=0;i<nums.size();i++) {
            pre += nums[i];
            if (mp.find(pre - k) != mp.end()){//在mp中找到了pre-k的值 
            	count += mp[pre - k];
			} 
            mp[pre]++;
        }
        return count;
    }

int main(){
	int n;
	int num;
	vector<int> nums;
	cin>>n;
	for(int i=0;i<n;i++){
		cin>>num;
		nums.push_back(num);
	}
	int k;
	cin>>k;
	int res=subarraySum(nums,k);
	cout<<res;
	return 0;
}