leetcode560.和为K的子数组

参考:
https://blog.csdn.net/tstsugeg/article/details/71107676

https://blog.csdn.net/haoxiaoxiaoyu/article/details/76341699

1.用数组sum记录下当前位置之前所有数字的相加和,这样下标[i, j)之间的数字之和就可以用sum[j]-sum[i]来计算

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int sz = nums.size(), cnt = 0;
        vector<int> sum(sz+1, 0);
        for(int i=0; i<sz; ++i) sum[i+1] = sum[i] + nums[i];//记录从0-i所有数的和
        for(int i=0; i<sz; ++i) {
            for(int j=i+1; j<=sz; ++j) {
                if(sum[j] - sum[i] == k) ++cnt; //遍历sum数组,找出所有值差为k的情况
            }
        }
        return cnt;
    }
};

2.每次以i为开始点(外循环),以j为结束点(内循环)开始搜索。与暴力搜索的不同之处在于这次是一边搜索一边求i-j的序列和 (了解一下即可)

public class Solution {
    public int subarraySum(int[] nums, int k) {
        int count = 0;
        for (int start = 0; start < nums.length; start++) {//start为开始点
            int sum=0;//当起点变化的时候,就从0开始重新计数
            for (int end = start; end < nums.length; end++) {//end为结束点
                sum+=nums[end]; //求start->end的和,下一个end只需要在前一个end和基础上加nums[]即可
                if (sum == k)
                    count++;
            }
        }
        return count;
    }
}

3.遍历数组同时建立一个哈希表,键为从0到当前位置的和sum,值为从0开始和为sum的位置个数,比如从0-3的和为5,从0到7的和为5,那么hash[5]=2.对于两个点i,j(i在j之前)来说,如果0到i的和为sum-k,0到j的和为sum,那么说明i-j的和为k。从j的角度来看,和为k的个数应该为hash[sum-k],即i之前所有和为sum-k的个数。因此我们在遍历数组的时候,除了同步建立哈希表之外,还需要用一个变量累加在当前位置之前的和为sum-k的个数。

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int sum = 0, cnt = 0;
        unordered_map<int, int> hash;       
        hash[0] = 1;
        for(auto n:nums) {
            sum += n; //累加
            cnt += hash[sum-k]; //当前和为sum,之前和为sum-k的个数
            ++hash[sum]; //键为和,值为个数
        }        
        return cnt;
    }
};

比如要找的和为10,当前和为8,在当前位置之前和为2的有3个,那么以当前位置结尾的和为8的区域就有3个。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值