資料結構
聯網
RDBMS
作業系統
Java
MS Excel
iOS
HTML
CSS
Android
Python
C 程式設計
C++
C#
MongoDB
MySQL
Javascript
PHP
物理
化學
生物
數學
英語
經濟學
心理學
社會學
時尚學
法學統計和為 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
更新於： 17-Feb-2020
186 次瀏覽
廣告