DP - 最大连续子序列和问题的变形
给定K个整数的序列{ N1, N2, …, NK },其任意连续子序列可表示为{ Ni, Ni+1, …, Nj },其中 1 <= i <= j <= K。最大连续子序列是所有连续子序列中元素和最大的一个,例如给定序列{ -2, 11, -4, 13, -5, -2 },其最大连续子序列为{ 11, -4, 13 },最大和为20。现在增加一个要求,即还需要输出该子序列的第一个和最后一个元素。
输入描述:
测试输入包含若干测试用例,每个测试用例占2行,第1行给出正整数K( K< 10000 ),第2行给出K个整数,中间用空格分隔。当K为0时,输入结束,该用例不被处理。
输出描述:
对每个测试用例,在1行里输出最大和、最大连续子序列的第一个和最后一个元素,中间用空格分隔。如果最大连续子序列不唯一,则输出序号i和j最小的那个(如输入样例的第2、3组)。若所有K个元素都是负数,则定义其最大和为0,输出整个序列的首尾元素。
测试自己代码是否通过:
https://www.nowcoder.com/questionTerminal/afe7c043f0644f60af98a0fba61af8e7
解题思路:
本题目既要找出最大连续子序列的最大和,又要找出最大子序列的左右端点,当存在多个结果时,返回第一种左右端点。
‘’’
解法一:
思路用dp[i],记录以i为末尾元素的子序列和。相对于 之前的,只要增加end去 记录最大子序列和的最后一个元素位置就可以,end确定后,从end往前找到一个小于0的位置,然后再+1,就是子序列开始,也就是begin的位置。
‘’’
def maxarray(nums):
#输入的数全小于 0 ,则输出最大子序列的的和为 0 ,序列的两端元素
flag = False
for item in nums:
if item > 0 :
flag = True
break
if flag == False:
print(0,nums[0],nums[-1])
#输入的数不全小于 0
dp = [0] * len(nums)
end ,i = 0,0
max_res = -1
if nums[0] <= 0:
dp[0] = 0
i += 1
else:
dp[0] = nums[0]
for i in range(1,len(nums)):
dp[i] = max(dp[i-1] + nums[i],nums[i])
if max_res < dp[i]:
end = i # 记录右端点
max_res = max(dp[i],max_res)
endval = nums[end]
flag = False
for i in range(end,0,-1):
if dp[i]< 0:
startval = nums[i+1]
flag = True
if flag == False :
startval = nums[0]
# print(startval,endval)
return max_res,startval,endval
def main():
k = int(input())
nums = list(map(int,input().split()[:k]))
# nums = [2,-1,2,-1,-1,3]
max_res,startval,endval = maxarray(nums)
print(max_res,startval,endval)
if __name__ == "__main__":
main()
解法二:
思路:dp[i]代表nums[i]结尾的最大子序列和,dp[i] = dp[i-1] + nums[i], 结合题目,若所有 K 个元素都是负数,则序列的最大和为 0 ,题目要求输出 序列最大和 与序列的首位元素。所以当dp[i] 小于 0 时,dp[i] 设为 0 。
def maxarray(nums):
#输入的数全小于 0 ,则输出最大子序列的的和为 0 ,序列的两端元素
flag = False
for item in nums:
if item > 0 :
flag = True
break
if flag == False:
print(0,nums[0],nums[-1])
#输入的数不全小于 0
dp = [0] * len(nums)
end,start = 0,0
max_res = -100000
if nums[0] <=0:
dp[0] = 0
start +=1
else:
dp[0] = nums[0]
max_res = nums[0]
start = 0
for i in range(1,len(nums)):
dp[i] = dp[i-1] + nums[i]
if dp[i] < 0 :
dp[i] = 0
start = i +1
if dp[i] > max_res:
max_res = dp[i]
end = i
return max_res,nums[start],nums[end]
def main():
k = int(input())
nums = list(map(int,input().split()[:k]))
# nums = [2,-1,2,-1,-2,3]
max_res,startval,endval = maxarray(nums)
print(max_res,startval,endval)
if __name__ == "__main__":
main()