mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-24 05:21:09 +00:00
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
"""
|
|
Given an array of integer elements and an integer 'k', we are required to find the
|
|
maximum sum of 'k' consecutive elements in the array.
|
|
|
|
Instead of using a nested for loop, in a Brute force approach we will use a technique
|
|
called 'Window sliding technique' where the nested loops can be converted to a single
|
|
loop to reduce time complexity.
|
|
"""
|
|
from typing import List
|
|
|
|
|
|
def max_sum_in_array(array: List[int], k: int) -> int:
|
|
"""
|
|
Returns the maximum sum of k consecutive elements
|
|
>>> arr = [1, 4, 2, 10, 2, 3, 1, 0, 20]
|
|
>>> k = 4
|
|
>>> max_sum_in_array(arr, k)
|
|
24
|
|
>>> k = 10
|
|
>>> max_sum_in_array(arr,k)
|
|
Traceback (most recent call last):
|
|
...
|
|
ValueError: Invalid Input
|
|
>>> arr = [1, 4, 2, 10, 2, 13, 1, 0, 2]
|
|
>>> k = 4
|
|
>>> max_sum_in_array(arr, k)
|
|
27
|
|
"""
|
|
if len(array) < k or k < 0:
|
|
raise ValueError("Invalid Input")
|
|
max_sum = current_sum = sum(array[:k])
|
|
for i in range(len(array) - k):
|
|
current_sum = current_sum - array[i] + array[i + k]
|
|
max_sum = max(max_sum, current_sum)
|
|
return max_sum
|
|
|
|
|
|
if __name__ == "__main__":
|
|
from doctest import testmod
|
|
from random import randint
|
|
|
|
testmod()
|
|
array = [randint(-1000, 1000) for i in range(100)]
|
|
k = randint(0, 110)
|
|
print(f"The maximum sum of {k} consecutive elements is {max_sum_in_array(array,k)}")
|