leetcode 560. Subarray Sum Equals K 、523. Continuous Subarray Sum、 325.Maximum Size Subarray Sum Equals k(lintcode 911)

整体上3个题都是求subarray,都是同一个思想,通过累加,然后判断和目标k值之间的关系,然后查看之前子数组的累加和。

map的存储:560题是存储的当前的累加和与个数

      561题是存储的当前累加和的余数与第一次出现这个余数的位置

      325题存储的是当前累加和与第一次出现这个和的位置

其实561与325都是求的最长长度,那就一定要存储的是第一次出现满足要求的位置,中间可能还出现这种满足要求的情况,但都不能进行存储

 

560. Subarray Sum Equals K

求和为k的连续子数组的个数

http://www.cnblogs.com/grandyang/p/6810361.html

 

如果当前的累加和减去k能找到值,一定是子数组。因为子数组一定是减去前面的值,累加和其实就是看到底减去几个值。

初始化的目的:如果当前位置和为k,那差值就是0,但是存储的值却是sum,不一定存储了0的值。

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        unordered_map<int,int> m;
        m[0] = 1;
        int sum = 0,res = 0;
        for(int i = 0;i < nums.size();i++){
            sum += nums[i];
            res += m[sum - k];
            m[sum]++;
        }
        return res;
    }
};

 

 

 

523. Continuous Subarray Sum

思路和560. Subarray Sum Equals K有些类似,因为都是找之前经过的数组的情况,只是560. Subarray Sum Equals K是和。一次循环,加每个数值,然后求余数,如果当前求余数在之前出现过,那当前的数减去之前的子数组的数就可以弥补。

错误解法:

如果一个数在hash_map中没有存储,比如10,使用m[10]不是不返回结果,而是返回0,这个0不代表value值,而是代表没有这个存储。这也是为什么下面这个代码在i = 2时就出错了,i=2时虽然hash_map中没有,但返回0,使得i - 0大于了1

class Solution {
public:
    bool checkSubarraySum(vector<int>& nums, int k) {
        unordered_map<int,int> m;
        m[0] = -1;
        int sum = 0;
        for(int i = 0;i < nums.size();i++){
            sum += nums[i];
            int tmp = (k == 0) ? sum : sum%k;
            if(i - m[tmp] > 1)
                return true;
            m[tmp] = i;
        }
        return false;
    }
};

 

注意1.这个地方m.count(tmp) == 1的时候不要去更新m[tmp] = i,因为这样不能得到最长的长度,去判断有没有至少2个的时候,应该用最远位置去判断。

  2. i - (i - 1) >= 1其实就能表示有两个数字,但是这里是需要减去之前的所有,所以必须是i - 1

正确解法:

class Solution {
public:
    bool checkSubarraySum(vector<int>& nums, int k) {
        unordered_map<int,int> m;
        m[0] = -1;
        int sum = 0;
        for(int i = 0;i < nums.size();i++){
            sum += nums[i];
            int tmp = (k == 0) ? sum : sum%k;
            if(m.count(tmp)){
                if(i - m[tmp] > 1)
                    return true;
            }
            else
                m[tmp] = i;
        }
        return false;
    }
};

 

325.Maximum Size Subarray Sum Equals k

这个题与523一样unordered_map中只需要保存第一个出现的位置,但保存的是sum,不是target,注意!

注意,你查找的时候是查找sum-k,如果查不到应该分为两种情况,即当前sum在map中有没有值。因为实际上这里是从左向右遍历,只要你找到第一个等于某个sum的,这个值就能成为你需要的最长的子数组。

class Solution {
public:
    /**
     * @param nums: an array
     * @param k: a target value
     * @return: the maximum length of a subarray that sums to k
     */
    int maxSubArrayLen(vector<int> &nums, int k) {
        // Write your code here
        unordered_map<int,int> m;
        m[0] = -1;
        int res = 0;
        int sum = 0;
        for(int i = 0;i < nums.size();i++){
            sum += nums[i];
            if(m.find(sum - k) != m.end()){
                res =  max(res,i - m[sum-k]);
            }
            else{
                if(m.find(sum) != m.end())
                    continue;
                else
                    m[sum] = i;
            }
        }
        return res;
    }
};

 

posted @ 2019-03-06 17:13  有梦就要去实现他  阅读(180)  评论(0编辑  收藏  举报