LeetCode560. Subarray Sum Equals K

我的方法：暴力。（结果：TLE。）

子数组的长度范围是1~n。这是最外层循环 i （0<i<length）。

对每一个长度，有一个开始，索引是 j (j+i<length)。

从 j 加到 j+i，又一个循环。

这是三重循环，我的。

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

 

---

1. 另一种暴力：

 最外层是开始的索引。

第二层是结束的索引。（也可以看成是子数组长度）。

第三层是累加。

public class Solution {
    public int subarraySum(int[] nums, int k) {
        int count = 0;
        for (int start = 0; start < nums.length; start++) {
            for (int end = start + 1; end <= nums.length; end++) {
                int sum = 0;
                for (int i = start; i < end; i++)
                    sum += nums[i];
                if (sum == k)
                    count++;
            }
        }
        return count;
    }
}

 

---

 

 2. 上面的暴力，第二层索引++，第三层又从头开始加。

其实2、3层可以合并成一个循环。

public class Solution {
    public int subarraySum(int[] nums, int k) {
        int count = 0;
        for (int start = 0; start < nums.length; start++) {
            int sum=0;
            for (int end = start; end < nums.length; end++) {
                sum+=nums[end];
                if (sum == k)
                    count++;
            }
        }
        return count;
    }
}

 

---

 

 

3. 累加和数组

借助一个sum数组。$\large sum[n]=\sum_{i=0}^{n-1}nums[i]$ ，其中sum[0]=0。

所以，子数组和nums[i:j]就是sum[j+1]-sum[i].

先一层循环，求出sum[]。

接下来一个两层循环，第一层是起始索引，第二层是结束。

 

---

 

4. 哈希

hashtable<sum , count>

一层循环，遍历到nums[i]时用sum-k，得到一个数。sum-k=tem , sum-tem=k. 

所以，还是累加和，不过不需要数组保存。用的是键值对。键是和，值是有多少个这样的和。（也可以看作有多少个索引，从开头加到这个索引，和为键）。查询为sum-k的键的值，就有多少个这样的索引，从这个索引到当前位置加起来和就为k‘，所以就有多少个子数组。

最开始要加个(0,1)进去，因为什么都还没加，和就是0。这个索引算作一个。

如果不加，那么sum=0时无法计数。

累加和数组法中sum[0]也是同样。

 

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

代码出自：http://www.cnblogs.com/grandyang/p/6810361.html

 代码有什么特殊的吗？有，才发现原来不需要判断key是否存在。不存在会返回0.

---

 

参考：

1. 官方题解

https://leetcode.com/problems/subarray-sum-equals-k/solution/