C++에서 하위 배열 합계는 K와 같습니다.

정수 배열과 정수 k가 있다고 가정하면 합이 k와 같은 연속 하위 배열의 총 수를 찾아야 합니다. 따라서 nums 배열이 [1, 1, 1]이고 k가 2이면 출력은 2가 됩니다.

이 문제를 해결하기 위해 다음 단계를 따릅니다. −

• sums, temp :=0, sums[0] :=1 및 ans :=0이라는 하나의 맵 정의
• 0 범위에서 배열 크기까지의 i에 대해
  • temp :=온도 + n[i]
  • 합계에 k – temp가 있으면
    • ans :=ans + sums[k - temp]
  • sum[-temp] 값을 1 증가
• 반환

예(C++)

더 나은 이해를 위해 다음 구현을 살펴보겠습니다. −

#include <bits/stdc++.h>
using namespace std;
class Solution {
public:
   int subarraySum(vector<int>& n, int k) {
      unordered_map <int, int> sums;
      int temp = 0;
      sums[0] = 1;
      int ans =0;
      for(int i =0;i<n.size();i++){
         temp+= n[i];
         if(sums.find(k-temp)!=sums.end()){
            ans += sums[k-temp];
         }
         sums[-temp]++;
      }
      return ans;
   }
};
main(){
   Solution ob;
   vector<int> v = {1,1,1};
   cout << (ob.subarraySum(v, 2));
}

입력

[1,1,1]
2

출력

2