wgQFVG - 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;
    }
    map<int,int>map1,map2;
    for(int i = 1 ; i<=n1 ; i++){
        presum[i] = presum[i-1]+A[i-1];
    }
    for(int i = 0 ; i<=n1 ; i++){
        if(map1.find(presum[i])==map1.end()){
            map1[presum[i]] = i;
        }
        map2[presum[i]] = i;
    }
    int maxi = 0,mini=INT_MAX;
    for(int i = 1 ; i<=n1 ; i++){
        if(map1.find(presum[i]-k)!=map1.end()){
            maxi = max(maxi,abs(i-map1[presum[i]-k]));
        }
        if(map2.find(presum[i]-k)!=map2.end()){
            mini = min(mini,abs(i-map2[presum[i]-k]));
        }
    }
    int cnt1 = 0,cnt2=0;
    for(int i = 1 ; i<=n1 ; i++){
        if(map1.find(presum[i]-k)!=map1.end()){
            if(maxi==(i-map1[presum[i]-k])){
                cnt1++;
            }
        }
        if(map2.find(presum[i]-k)!=map2.end()){
            if(mini==(i-map2[presum[i]-k])){
                cnt2++;
            }
        }
    }
    cout<<cnt2<<" "<<cnt1<<endl;
	return 0;
}