首页 > 其他 > 详细

【leetcode】560. Subarray Sum Equals K

时间:2018-04-14 12:22:18      阅读:195      评论:0      收藏:0      [点我收藏+]

题目如下:技术分享图片解题思路:本题的关键在于题目限定了是连续的数组,我们用一个dp数组保存第i位到数组末位的和。例如nums = [1,1,1],那么dp = [3,2,1], dp[i]表示nums[i]+nums[i+1] +...+nums[len(nums)-1],有了这一个dp数组后,我们很容易就可以得到递推表达式 sum(i,j) = dp[i] - dp[j+1]。最后,顺序遍历dp数组,对于任意的dp[i],只要找到对应的dp[k-i]就可以了。

代码如下:

class Solution(object):
    def subarraySum(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        dp = [0 for x in nums]
        dp[-1] = nums[-1]
        dic = {}
        dic[dp[-1]] = 1
        for i in xrange(-2,-len(nums)-1,-1):
            dp[i] = dp[i+1] + nums[i]
            if dic.has_key(dp[i]):
                dic[dp[i]] += 1
            else:
                dic[dp[i]] = 1
        res = 0
        for i,v in enumerate(dp):
            if v == k:
                res += 1
            dic[v] -= 1
            if dic.has_key(v-k):
                res += dic[v-k]

        return res

 

【leetcode】560. Subarray Sum Equals K

原文:https://www.cnblogs.com/seyjs/p/8830994.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!