Function to return the number of subarrays of the given array such that the remainder when dividing the sum of its elements by K is equal to the number of its elements
def sub_arrays ( a , n , k ) : NEW_LINE