Day6.和为k的子数组--LeeCode练习

题目分析

给定一个整数数组和一个整数k,你需要找出连续子数组中和为k的数组个数。

代码1

public:
    int subarraySum(vector<int>& nums, int k) {
        int p=0;
        for(int i=0;i<nums.size();i++){
            int sum=nums[i];
            if(sum==k){
                p++;
            }
            for(int j=i+1;j<nums.size();j++){
                sum+=nums[j];
                if(sum==k){
                    p++;
                }
            }
        }
        return p;
    }
};

这里用了两个for循环,使用了暴力解法,计算出所有的连续子数组的结果,最后找出所有符合的子数组。

代码2

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        unordered_map<int,int> mp;
        mp[0]=1;//相当于从头开始的一串数组
        int p=0,q=0;
        for(const auto& x:nums){
            p+=x;
            if(mp.find(p-k)!=mp.end()) q+=mp[p-k];
            mp[p]++;
        }
        return q;
    }
};

题目和代码2来自于这里
时间复杂度:O(n),空间复杂度:O(n)
这里有十分详细的算法思想

你可能感兴趣的:(Day6.和为k的子数组--LeeCode练习)