Python: максимальная сумма подмассива
Я только что попытался начать изучать Python сегодня, поэтому я не разбираюсь в его функциях. Тем не менее, я нашел максимальную проблему с подмассивом и хотел попытаться решить ее с помощью нескольких простых логических команд, имеющихся в моем распоряжении. Я застрял и почти уверен, что проблема в моей логике, а не в синтаксисе, хотя я вполне могу ошибаться. Вот мой код пока...
def maxSequence(arr): #Latest attempt, some issue.
old_arr = []
print(arr)
while old_arr != arr:
old_arr = arr
if arr[0] > 0 and arr[len(arr)-1] > 0: #For when both sides are positive, need to be sure there is not anything to be gained by eliminating some side section
new_sum = 0
y=0
while new_sum >= 0 and y != -1:
new_sum = sum(arr[:y])
y=y+1
if y == len(arr)-1:
y=-1
if y != -1:
arr = arr[y-1:]
print("left %s" %(new_sum))
print("left %s" % (arr))
new_sum = 0
y = 0
while new_sum >= 0 and y != -1:
new_sum=sum(arr[(len(arr)-y):])
y=y+1
if y == len(arr)-1:
y=-1
if y != -1:
arr = arr[:len(arr)-y+1]
print("right %s" %(new_sum))
print("right %s" % (arr))
else:
while arr[0] < 0 or arr[len(arr)-1] < 0: #To eliminate negatives on sides
if arr[0] < 0:
arr = arr[1:]
if arr[len(arr)-1] < 0:
arr = arr[:len(arr)-1]
print("negative %s" % (arr))
print(arr)
print(sum(arr))
Различные функции печати показывают каждое решение, принятое программой, и помогают мне визуализировать, что происходит при выполнении циклов.
Это не дает правильный результат 105, когда дан список
[26, -25, -23, -2, 3, -6, -5, 15, 7, 8, 17, 15, 29, -2, 16, -25, -8, -25, -27, 15, -29, -11, -12, 1, -14, 21, 20, 30, -29, 17, 9, -19, 28, 11, 6, -10, -25, 27, -18, 6, -13, -13, 25, -22, 8, 9, -4, -25, 17, -26]
В итоге получается сумма 94, сократив список до
[21, 20, 30, -29, 17, 9, -19, 28, 11, 6]
Извините за длинный пост, но я просто не могу понять, что я делаю неправильно. Заранее спасибо за помощь!
Вот результат, когда вышеупомянутый список предоставляется в качестве входных данных, я прошел и рассмотрел каждый шаг, и каждое исключение из списка кажется мне логичным, я не знаю, как можно получить конечную сумму 105. Если бы кто-нибудь мог помочь мне понять, я был бы очень признателен!
[26, -25, -23, -2, 3, -6, -5, 15, 7, 8, 17, 15, 29, -2, 16, -25, -8, -25,
-27, 15, -29, -11, -12, 1, -14, 21, 20, 30, -29, 17, 9, -19, 28, 11, 6, -10,
-25, 27, -18, 6, -13, -13, 25, -22, 8, 9, -4, -25, 17, -26]
negative [26, -25, -23, -2, 3, -6, -5, 15, 7, 8, 17, 15, 29, -2, 16, -25,
-8, -25, -27, 15, -29, -11, -12, 1, -14, 21, 20, 30, -29, 17, 9, -19, 28,
11, 6, -10, -25, 27, -18, 6, -13, -13, 25, -22, 8, 9, -4, -25, 17]
left -22
left [-2, 3, -6, -5, 15, 7, 8, 17, 15, 29, -2, 16, -25, -8, -25, -27, 15,
-29, -11, -12, 1, -14, 21, 20, 30, -29, 17, 9, -19, 28, 11, 6, -10, -25, 27,
-18, 6, -13, -13, 25, -22, 8, 9, -4, -25, 17]
right -8
right [-2, 3, -6, -5, 15, 7, 8, 17, 15, 29, -2, 16, -25, -8, -25, -27, 15,
-29, -11, -12, 1, -14, 21, 20, 30, -29, 17, 9, -19, 28, 11, 6, -10, -25, 27,
-18, 6, -13, -13, 25, -22, 8, 9, -4]
negative [3, -6, -5, 15, 7, 8, 17, 15, 29, -2, 16, -25, -8, -25, -27, 15,
-29, -11, -12, 1, -14, 21, 20, 30, -29, 17, 9, -19, 28, 11, 6, -10, -25, 27,
-18, 6, -13, -13, 25, -22, 8, 9]
left -3
left [-5, 15, 7, 8, 17, 15, 29, -2, 16, -25, -8, -25, -27, 15, -29, -11,
-12, 1, -14, 21, 20, 30, -29, 17, 9, -19, 28, 11, 6, -10, -25, 27, -18, 6,
-13, -13, 25, -22, 8, 9]
right -5
right [-5, 15, 7, 8, 17, 15, 29, -2, 16, -25, -8, -25, -27, 15, -29, -11,
-12, 1, -14, 21, 20, 30, -29, 17, 9, -19, 28, 11, 6, -10, -25, 27, -18, 6,
-13, -13, 25]
negative [15, 7, 8, 17, 15, 29, -2, 16, -25, -8, -25, -27, 15, -29, -11,
-12, 1, -14, 21, 20, 30, -29, 17, 9, -19, 28, 11, 6, -10, -25, 27, -18, 6,
-13, -13, 25]
left -5
left [-12, 1, -14, 21, 20, 30, -29, 17, 9, -19, 28, 11, 6, -10, -25, 27,
-18, 6, -13, -13, 25]
right -1
right [-12, 1, -14, 21, 20, 30, -29, 17, 9, -19, 28, 11, 6, -10, -25, 27,
-18, 6]
negative [1, -14, 21, 20, 30, -29, 17, 9, -19, 28, 11, 6, -10, -25, 27, -18,
6]
left -13
left [21, 20, 30, -29, 17, 9, -19, 28, 11, 6, -10, -25, 27, -18, 6]
right -12
right [21, 20, 30, -29, 17, 9, -19, 28, 11, 6, -10, -25, 27]
left 84
left [21, 20, 30, -29, 17, 9, -19, 28, 11, 6, -10, -25, 27]
right -8
right [21, 20, 30, -29, 17, 9, -19, 28, 11, 6]
left 77
left [21, 20, 30, -29, 17, 9, -19, 28, 11, 6]
right 53
right [21, 20, 30, -29, 17, 9, -19, 28, 11, 6]
[21, 20, 30, -29, 17, 9, -19, 28, 11, 6]
94
4 ответа
Я продемонстрирую, что не так с вашим алгоритмом, на небольшом примере.
Скажите, что вход
[2, -3, 1]
[2] явно максимальный подмассив. Тем не менее, ваш алгоритм будет смотреть на эту часть:
[2, -3, 1]
^^^^^
и увидеть, что он может увеличить сумму массива, удалив [2, -3],
Удаление отрицательных подмассивов с концов делает неправильную вещь, если максимальный подмассив является частью отрицательного подмассива. Вам понадобится более умный алгоритм.
Почему бы просто не использовать алгоритм Кадане, который занимает O(n) времени??
Алгоритм Кадане (DP):
Initialize:
max_so_far = 0
max_ending_here = 0
Loop for each element of the array
(a) max_ending_here = max_ending_here + a[i]
(b) if(max_ending_here < 0)
max_ending_here = 0
(c) if(max_so_far < max_ending_here)
max_so_far = max_ending_here
return max_so_far
def contiguous(list_val, size):
if size<=len(list_val):
iterations = len(list_val)-size+1
count = 0
num = []
for i in range(iterations):
sum_val = 0
for j in range(count,size):
sum_val+=list_val[j]
num.append(sum_val)
count +=1
size +=1
print(max(num))
else:
print("Error: given size {} is greater than length of list".format(size))
list_val = input("Enter list: ")
size = input("Enter the pos: ")
contiguous(list_val, size)
попробуйте этот код. это работает с временной сложностью O(n), потому что он просто запускается n раз.
def manSubArraySum(a):
currentbest = 0
overallbest = a[0]
for p in a: # [1,2,-2,4]
currentbest = currentbest+p # first step cb = -10
if currentbest> overallbest: # !true
overallbest = currentbest
if currentbest<0: # nottrue
currentbest =0 # 0
return overallbest