สมมติว่าเรามีอาร์เรย์ A ของจำนวนเต็ม เราต้องหาจำนวนของ subarray ที่ไม่ว่างที่อยู่ติดกัน ซึ่งมีผลรวมหารด้วย k ลงตัว ถ้า A =[4,5,0,-2,-3,1] และ k =5 ผลลัพธ์จะเป็น 7 มีอาร์เรย์ย่อยเจ็ดชุด [[4,5,0,-2,-3,1], [5], [5,0],[5,0,-2,-3], [0], [0,-2,- 3], [-2,-3]]
เพื่อแก้ปัญหานี้ เราจะทำตามขั้นตอนเหล่านี้ -
- สร้างหนึ่งแผนที่ m และตั้งค่า m[0] เป็น 1
- อุณหภูมิ :=0, ans :=0, และ n :=ขนาดของอาร์เรย์ a
- สำหรับ i ในช่วง 0 ถึง n – 1
- temp :=temp + a[i]
- x :=(ตัวดัดแปลงชั่วคราว k + k) mod k
- ans :=ans + m[x]
- เพิ่ม m[x] ขึ้น 1
- คืนสินค้า
ให้เราดูการใช้งานต่อไปนี้เพื่อความเข้าใจที่ดีขึ้น -
ตัวอย่าง
#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
