|
| 1 | +""" |
| 2 | +Given a array of length n, max_sub_array_sum() finds the maximum of sum of contiguous sub-array using divide and conquer method. |
| 3 | +
|
| 4 | +Time complexity : O(n log n) |
| 5 | +
|
| 6 | +Ref : INTRODUCTION TO ALGORITHMS THIRD EDITION (section : 4, sub-section : 4.1, page : 70) |
| 7 | +
|
| 8 | +""" |
| 9 | + |
| 10 | + |
| 11 | +def max_sum_from_start(array): |
| 12 | + |
| 13 | + """ This function finds the maximum contiguous sum of array from 0 index |
| 14 | +
|
| 15 | + Parameters : |
| 16 | + array (list[int]) : given array |
| 17 | + |
| 18 | + Returns : |
| 19 | + max_sum (int) : maximum contiguous sum of array from 0 index |
| 20 | +
|
| 21 | + """ |
| 22 | + array_sum = 0 |
| 23 | + max_sum = float("-inf") |
| 24 | + for num in array: |
| 25 | + array_sum += num |
| 26 | + if array_sum > max_sum: |
| 27 | + max_sum = array_sum |
| 28 | + return max_sum |
| 29 | + |
| 30 | + |
| 31 | +def max_cross_array_sum(array, left, mid, right): |
| 32 | + |
| 33 | + """ This function finds the maximum contiguous sum of left and right arrays |
| 34 | +
|
| 35 | + Parameters : |
| 36 | + array, left, mid, right (list[int], int, int, int) |
| 37 | + |
| 38 | + Returns : |
| 39 | + (int) : maximum of sum of contiguous sum of left and right arrays |
| 40 | +
|
| 41 | + """ |
| 42 | + |
| 43 | + max_sum_of_left = max_sum_from_start(array[left:mid+1][::-1]) |
| 44 | + max_sum_of_right = max_sum_from_start(array[mid+1: right+1]) |
| 45 | + return max_sum_of_left + max_sum_of_right |
| 46 | + |
| 47 | + |
| 48 | +def max_sub_array_sum(array, left, right): |
| 49 | + |
| 50 | + """ This function finds the maximum of sum of contiguous sub-array using divide and conquer method |
| 51 | +
|
| 52 | + Parameters : |
| 53 | + array, left, right (list[int], int, int) : given array, current left index and current right index |
| 54 | + |
| 55 | + Returns : |
| 56 | + int : maximum of sum of contiguous sub-array |
| 57 | +
|
| 58 | + """ |
| 59 | + |
| 60 | + """ base case: |
| 61 | + array has only one element""" |
| 62 | + if left == right: |
| 63 | + return array[right] |
| 64 | + |
| 65 | + """ Recursion""" |
| 66 | + |
| 67 | + mid = (left + right) // 2 |
| 68 | + left_half_sum = max_sub_array_sum(array, left, mid) |
| 69 | + right_half_sum = max_sub_array_sum(array, mid + 1, right) |
| 70 | + cross_sum = max_cross_array_sum(array, left, mid, right) |
| 71 | + return max(left_half_sum, right_half_sum, cross_sum) |
| 72 | + |
| 73 | + |
| 74 | +array = [-2, -5, 6, -2, -3, 1, 5, -6] |
| 75 | +array_length = len(array) |
| 76 | +print("Maximum sum of contiguous subarray:", max_sub_array_sum(array, 0, array_length - 1)) |
| 77 | + |
0 commit comments