Computer >> 컴퓨터 >  >> 프로그램 작성 >> C++

k로 나눌 수 있는 합을 갖는 모든 하위 배열을 셉니다.

<시간/>

이 튜토리얼에서는 합이 k로 나누어지는 하위 배열의 수를 찾는 프로그램에 대해 논의할 것입니다.

이를 위해 배열과 값 k가 제공됩니다. 우리의 임무는 합이 주어진 값 k와 같은 하위 배열의 수를 찾는 것입니다.

#include <bits/stdc++.h>
using namespace std;
//counting subarrays with k sum
int count_subarray(int arr[], int n, int k){
   int mod[k];
   memset(mod, 0, sizeof(mod));
   int cumSum = 0;
   for (int i = 0; i < n; i++) {
      cumSum += arr[i];
      //taking modulus to get positive sum
      mod[((cumSum % k) + k) % k]++;
   }
   int result = 0;
   for (int i = 0; i < k; i++)
      if (mod[i] > 1)
         result += (mod[i] * (mod[i] - 1)) / 2;
   result += mod[0];
   return result;
}
int main(){
   int arr[] = { 4, 5, 0, -2, -3, 1 };
   int k = 5;
   int n = sizeof(arr) / sizeof(arr[0]);
   cout << count_subarray(arr, n, k) << endl;
   int arr1[] = { 4, 5, 0, -12, -23, 1 };
   nt k1 = 5;
   int n1 = sizeof(arr1) / sizeof(arr1[0]);
   cout << count_subarray(arr1, n1, k1) << endl;
   return 0;
}

출력

7
7