Advertisement
ekzolot

Untitled

Sep 5th, 2022
102
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.64 KB | None | 0 0
  1. #include <iostream>
  2. #include <vector>
  3. #include <unordered_set>
  4. using namespace std;
  5. int main(){
  6.     int n, k;
  7.     cin>>n>>k;
  8.     vector <long long> a(n);
  9.     vector <long long> prefix(n);
  10.     for (int i=0; i<n; i++){
  11.         cin>>a[i];
  12.     }
  13.     prefix[0]=a[0];
  14.     for (int i=1; i<n; i++){
  15.         prefix[i]=prefix[i-1]+a[i];
  16.     }
  17.     unordered_set <long long> b(n);
  18.     for (int i=0; i<n; i++){
  19.         b.insert(prefix[i]);
  20.     }
  21.     long long answer=0;
  22.     for (auto x:prefix){
  23.         if(x==k){
  24.             answer++;
  25.         }
  26.         if (b.count(x-k)==1){
  27.             answer++;
  28.         }
  29.     }
  30.     cout<<answer<<"\n";
  31. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement