刷題560. Subarray Sum Equals K

一、題目說明

題目560. Subarray Sum Equals K，給一列整數和一個整數k，找到所有子數組和是k的個數。難度是Medium！

二、我的解答

這個題目第一印象是用dp，很快代碼就寫好了。然而 Memory Limit Exceeded，其中vector<vector<int> > dp(len+1,vector<int>(len+1,0));

代碼如下：

class Solution{
	public:
		int subarraySum(vector<int>& nums,int k){
			int len = nums.size();
			int total = 0;
			vector<vector<int> > dp(len+1,vector<int>(len+1,0));
			for(int i=0;i<len;i++){
				dp[i][i] = nums[i];
				if(nums[i]==k){
					total++;
				}
			}
			for(int i=0;i<len;i++){
				for(int j=i+1;j<len;j++){
					dp[i][j] = dp[i][j-1] + nums[j];
					if(dp[i][j]==k){
						total++;
					}
				}
			}
			return total;
		}
};

三、優化措施

那就一個一計算吧，用dfs:

class Solution{
	public:
		int subarraySum(vector<int>& nums,int k){
			int len = nums.size();
			int total = 0;

			for(int i=0;i<len;i++){
				total += dfs(nums,i,k);
			}
			
			return total;
		}
		int dfs(vector<int>& nums,int start,int k){
			int total = 0;
			int sum = 0;
			for(int i=start;i<nums.size();i++){
				sum += nums[i];
				if(sum==k) total++;
			}
			return total;
		}
};

性能如下：

Runtime: 464 ms, faster than 24.82% of C++ online submissions for Subarray Sum Equals K.
Memory Usage: 9.8 MB, less than 98.67% of C++ online submissions for Subarray Sum Equals K.