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

更新時間:2020-4-30

464 次瀏覽

開啟您的 職業生涯

完成課程後獲得認證

開始學習
廣告
© . All rights reserved.