Subarrays with sum K

Medium Accuracy: 65.67% Submissions: 5548 Points: 4

Given an unsorted array of integers, find the number of continuous subarrays having sum exactly equal to a given number k.


Example 1:

Input:
N = 5
Arr = {10 , 2, -2, -20, 10}
k = -10
Output: 3
Explaination: 
Subarrays: arr[0...3], arr[1...4], arr[3..4]
have sum exactly equal to -10.


Example 2:

Input:
N = 6
Arr = {9, 4, 20, 3, 10, 5}
k = 33
Output: 2
Explaination: 
Subarrays : arr[0...2], arr[2...4] have sum
exactly equal to 33.


Your Task:
You don't need to read input or print anything. Your task is to complete the function findSubArraySum() which takes the array Arr[] and its size N and k as input parameters and returns the count of subarrays.



from collections import defaultdict


class Solution:

    def findSubArraySum(self, Arr, n, k):

        # code here

        d = defaultdict(lambda :0)

        res = 0

        c = 0

        for i in Arr:

            c +=i

            #print(c,'csum')

            #print(c-k,'prevsum')

            if(c==k):

                res+=1

            if c-k in d:

                res+=d[c-k]

            d[c]+=1

            #print(res,'res')

        return res

                


Comments

Post a Comment

Popular posts from this blog

Large Factorial of array

You are given an array   A   of integers of length   N . You need to calculate factorial of each number. The answer can be very large, so print it modulo   10 9  + 7 .   Example 1: Input: N = 5 A[] = {0, 1, 2, 3, 4} Output: 1 1 2 6 24 Explanation: Factorial of 0 is 1, factorial of 1 is 1, factorial of 2 is 2,  factorial of 3 is 6 and so on. Example 2: Input: N = 3 A[] = {5, 6, 3} Output: 120 720 6 Explanation: Factorial of 5 is 120, factorial of 6 is 720,  factorial of 3 is 6. Your Task: Complete the function  factorial()  which takes   list  a  and single integer  n , as input parameters and returns a list of factorials of the numbers in the list a.
Read more

LEARN SOMMETHING INTRESTING

  Yeah, I love Python one-liners. Let us break it down. input()   This is used to fetch input from the user. In this case. We are expecting the user to provide a list of integers. How do I know? Look next. input().strip()   This will eliminate trailing spaces from the user input, if they are there. Why is it required? Because some users will do that and that might break your program. Remember that input() will cast the input as string. So, now we have a string of integers like so: “1 2 4 42” input.strip().split()   split() is used to create a Python list out of a string. If no delimiter is given, this breaks the string by spaces. So, now we have: [“1”, “2”, “4”, “42”] map(int, input().strip().split())   map() takes two arguments. The first one is the method to apply, the second one is the data to apply it to. By this understanding, we can see this is doing nothing but typecasting every element of the list to an integer value. Since map returns the data type it w...
Read more

Value equal to index value

Given an array   Arr   of   N   positive integers. Your task is to find the elements whose value is equal to that of its index value. Example 1: Input: N = 5 Arr[] = {15, 2, 45, 12, 7} Output: 2 Explanation: Only Arr[2] = 2 exists here. Example 2: Input: N = 1 Arr[] = {1} Output: 1 Explanation: Here Arr[1] = 1 exists. Your Task:   You don't need to read input or print anything. Your task is to complete the function  valueEqualToIndex()  which takes the array of integers  arr[]   and  n  as parameters and returns an array of indices where the given conditions are satified. When there is not such element exists then return an empty array of length 0. Expected Time Complexity:  O(N) Expected Auxiliary Space:  O(1) Note : There can be more than one element in the array which have same value as their index. You need to include every such element's index. Follows 1-based indexing of the array. Constraints: ...
Read more