THd9Ac - Online C++0x Compiler & Debugging Tool

#include <bits/stdc++.h>
using namespace std;

int main() {
	int n1,k;
    cin>>n1>>k;
    int A[n1];
    for(int i = 0 ; i<n1 ; i++){
        cin>>A[i];
    }
    int presum[n1+1];
    for(int i = 0 ; i<n1+1 ; i++){
        presum[i] = 0;
    }
    for(int i = 1 ; i<=n1 ; i++){
        presum[i] = presum[i-1]+A[i-1];
    }
    map<int,int>map;
    int cnt = 0;
    for(int i = 1 ; i<=n1 ; i++){
        if(presum[i]==k){
           cnt++;
        }
        if(map.find(presum[i]-k)!=map.end()){
            cnt += map[presum[i]-k];
        }
        map[presum[i]]++;
    }
    cout<<cnt<<endl;
	return 0;
}