C++ 中可被 K 整除的子数组和
假设我们有一个由整数组成的数组 A。我们需要找出连续的非空子数组数量,这些子数组的和可以被 k 整除。如果 A = [4,5,0,-2,-3,1] 并且 k = 5,则输出将会是 7。共有 7 个子数组。[[4,5,0,-2,-3,1], [5], [5,0],[5,0,-2,-3], [0], [0,-2,-3], [-2,-3]]
要解决这个问题,我们将按照以下步骤进行操作:
- 创建一个名为 m 的 Map,并将 m[0] 设置为 1
- temp := 0, ans := 0 并且 n := 数组 a 的大小
- 对于 0 至 n – 1 范围内的 i
- temp := temp + a[i]
- x := (temp mod k + k) mod k
- ans := ans + m[x]
- 将 m[x] 增加 1
- 返回 ans
下面让我们查看以下实现,以获得更深入的了解:
示例
#include <bits/stdc++.h> using namespace std; class Solution { public: int subarraysDivByK(vector<int>& a, int k) { unordered_map <int, int> m; m[0] = 1; int temp = 0; int ans = 0; int n = a.size(); for(int i = 0; i < n; i++){ temp += a[i]; int x = (temp % k + k) % k; ans += m[x]; m[x]++; } return ans; } }; main(){ vector<int> v = {4,5,0,-2,-3,1}; Solution ob; cout <<(ob.subarraysDivByK(v, 5)); }
输入
[4,5,0,-2,-3,1] 5
输出
7
广告