数组查询
问题描述
Description
Given an array, the task is to complete the function which finds the maximum sum subarray, where you may remove at most one element to get the maximum sum.
Input
第一行为测试用例个数T;后面每两行表示一个用例,第一行为用例中数组长度N,第二行为数组具体内容。
Output
每一行表示对应用例的结果。
Sample Input 1
1
5
1 2 3 -4 5
Sample Output 1
11
Hint
例如,对一个数组A[] = {1, 2, 3, -4, 5},要移除-4得到最大和的子数组,和为11.
题目解析
- 数组最多删掉一个元素
- 求该数组的连续子数组和的最大值
注意:
- 数组全为负数[-1,-2,-3,-4] ,此时输出-1
- 连续子数组是可以包含负数的,假设删除一个元素后的数组为[-2,1,-3,4,-1,2,1,-5,4],那么连续子数组 [4,-1,2,1] 的和最大,为 6。
- 数组全为正数,就不要删除了
思路解析
思路1
如果这道题没有要求可以删掉一个元素的话,那么这道题就是leetcode上的一道题目,leetcode:最大子序和,因此想到了一种很暴力的方法,如果可以删掉一个元素的话,那么就分别将每个元素都删掉,获得一个新数组,然后统计一次新数组的子数组的和,最后输出最长的长度,这样时间复杂度是O(n2)
思路2
使用dp
代码实现(python)
思路1:
暴力方法,时间复杂度是O(n2)
def maxSumSubarray(arr, n):
all_arr = arr
# print(sum(all_arr))
max_total = min(arr)
if max_total >= 0:
print(sum(arr))
else:
for temp in all_arr:
arr = all_arr.copy()
arr.remove(temp) # 这里用remove不会移除对应位置的元素,歪打正着移除后值不变
dp = [0] * len(arr)
total = 0
for i in range(len(arr)): # 到当前这个数最长的字符串
total += arr[i]
max_total = max(total, max_total)
if total >= 0:
dp[i] = total
else:
total = 0
dp[i] = 0
# print(max_total)
print(max_total)
if __name__ == '__main__':
# all_arr = [1, 2, 3, -4, -3, 5]
for _ in range(int(input())):
n = int(input())
all_arr = list(map(int, input().strip().split()))
maxSumSubarray(all_arr, n)
思路2:
dp
def maxSumSubarray():
fw = [arr[i] for i in range(n)]
bw = [arr[i] for i in range(n)]
for i in range(1,n):
fw[i] = max(fw[i],fw[i]+fw[i-1])
for i in range(n-2,-1,-1):
bw[i] = max(bw[i],bw[i]+bw[i+1])
ans = arr[0]
for i in range(n):
if fw[i] == bw[i]:
ans = max(ans,fw[i])
else:
ans = max(ans,fw[i]+bw[i]-(2*arr[i]),fw[i]+bw[i]-arr[i])
return(ans)
# Driver Program
if __name__=='__main__':
t = int(input())
for i in range(t):
n = int(input())
arr = list(map(int, input().strip().split()))
print(maxSumSubarray(arr, n))
# Contributed By: Harshit Sidhwa