1.和为 K 的子数组
给你一个整数数组 nums 和一个整数 k ,请你统计并返回 该数组中和为 k 的连续子数组的个数 。
子数组是数组中元素的连续非空序列。
方法一:前缀和
#方法一:前缀和
def subarraySum(nums,k):
dic={0:1} #记录出现过前缀和的次数
cur_sum,res = 0,0 #记录当前前缀和
for i in range(len(nums)):
cur_sum+=nums[i]
tem=cur_sum-k #计算需要减去的前缀和
if tem in dic:
res+=dic[tem] #看需要减去的前缀和出现过多少次
if cur_sum in dic:
dic[cur_sum]+=1
else:
dic[cur_sum]=1
return res
2.最大平均值和的分组
给定数组 nums 和一个整数 k 。我们将给定的数组 nums 分成 最多 k 个非空子数组,且数组内部是连续的 。 分数 由每个子数组内的平均值的总和构成。
注意我们必须使用 nums 数组中的每一个数进行分组,并且分数不一定需要是整数。
返回我们所能得到的最大 分数 是多少。答案误差在 10-6 内被视为是正确的。
方法一:动态规划+前缀和
#方法一:动态规划+前缀和
def largestSumOfAverages(nums,k):
dp=[[0.0]*(k+1) for i in range(len(nums)+1)] #dp[i][j] 表示前i个元素分成k组时的最大平均和分组
prefix=[0.0 for i in range(len(nums) +1)] #构建前缀和数组
for i in range(1,len(nums) +1):
prefix[i]=prefix[i-1]+nums[i-1]
for i in range(1,len(nums) +1):
dp[i][1]=prefix[i]/i #dp[i][1]表示只能分成一组,所以就是前缀和的平均值
for j in range(2,k+1):
for i in range(j,len(nums) +1): #i>=j ,x>=j-1
for x in range(j-1,i): #dp[5][3]=max(dp[2][2]+()/3,dp[3][2]+()/2,dp[4][2]+()/1)
dp[i][j]=max(dp[i][j],dp[x][j-1]+(prefix[i]-prefix[x])/(i-x))
return dp[len(nums)][k]
3.和相同的二元子数组
给你一个二元数组 nums ,和一个整数 goal ,请你统计并返回有多少个和为 goal 的 非空 子数组。
子数组 是数组的一段连续部分。
方法一:前缀和+哈希表
#方法一:前缀和+哈希表
def numSubarraysWithSum(nums,goal):
presum=[0 for i in range(len(nums)+1)] #构建前缀和数组
dic,res={0:1},0
for i in range(1,len(nums)+1):
presum[i]=presum[i-1]+nums[i-1] #前缀和数组
for i in range(1,len(nums) +1):
tem=presum[i]-goal #获得需要的前缀和大小
if tem in dic:
res+=dic[tem] #tem出现的次数即为可能的结果数目
if presum[i] not in dic:
dic[presum[i]]=1 #记录当前前缀和出现的次数
else:
dic[presum[i]]+=1
return res