Python program to find number of m contiguous elements of a List with a given sum
Given a list ‘L’, a sum ‘S’ and number of elements to take at a time ‘m’. The task is to find how many ways sum s can be found by adding any m contiguous elements. Examples:
Input :
1 2 1 3 2
3 2
Output :
2
Input :
1 1 1 1 1 1
3 2
Output :
0
For example 1, we have to find a sum 3 with the help of any 2 contiguous elements of the list. This can be done in two ways 1+2 and 2+1 so answer is 2 Method 1: (Simple) Starting from beginning we select first m elements and find their sum if sum = s then increment the count by one, and increment index position by one select next m elements and repeat the same until the whole list is covered.
Python3
def SearchWay(l, s, m):
sum1 = 0
count = 0
for i in range ( len (l) - m + 1 ):
for j in range (i, i + m):
sum1 + = l[j]
if sum1 = = s:
count + = 1
sum1 = 0
return count
l = [ 1 , 2 , 1 , 3 , 2 ]
s = 3
m = 2
ans = SearchWay(l, s, m)
print (ans)
|
Output:
2
Time complexity: O(n^2) in worst case. Method 2: (Efficient) Initialize a variable index as 0 and curr_sum as the first element. index indicates the count of contiguous elements and curr_sum indicates the sum of the current index elements. Start from the second element and add all elements one by one to the curr_sum. If curr_sum becomes equal to the sum and index becomes equal to m, then increment the value of count by 1. If index exceeds the m, then remove trailing elements and initialize index as m-1.
Python3
def SearchWay(l, s, m):
n = len (l)
count = start = 0
curr_sum = l[ 0 ]
index = 1
i = 1
while i < = n:
if curr_sum = = s and index = = m:
count + = 1
while index > = m:
curr_sum - = l[start]
start + = 1
index = m - 1
if i < n:
curr_sum + = l[i]
index + = 1
i + = 1
return count
l = [ 1 , 2 , 1 , 3 , 2 ]
s = 3
m = 2
ans = SearchWay(l, s, m)
print (ans)
|
Output:
2
The time complexity of method 2 looks more than O(n), but if we take a closer look at the program, then we can figure out the time complexity is O(n). We can prove it by counting the number of operations performed on every element of arr[] in the worst case. There are at most 2 operations performed on every element: (a) the element is added to the curr_sum (b) the element is subtracted from curr_sum. So the upper bound on the number of operations is 2n which is O(n).
Last Updated :
24 Mar, 2022
Like Article
Save Article
Share your thoughts in the comments
Please Login to comment...