<백준> 2003

수들의 합 2

문제
N개의 수로 된 수열 A[1], A[2], …, A[N] 이 있다. 이 수열의 i번째 수부터 j번째 수까지의 합 A[i] + A[i+1] + … + A[j-1] + A[j]가 M이 되는 경우의 수를 구하는 프로그램을 작성하시오.

입력
첫째 줄에 N(1 ≤ N ≤ 10,000), M(1 ≤ M ≤ 300,000,000)이 주어진다. 다음 줄에는 A[1], A[2], …, A[N]이 공백으로 분리되어 주어진다. 각각의 A[x]는 30,000을 넘지 않는 자연수이다.

출력
첫째 줄에 경우의 수를 출력한다.

투포인터를 사용해서 해결했다.

int main()
{
    int n, m;
    cin >> n >> m;    

    int arr[10001] = { 0, };
    //vector<int> arr(n);

    for (int i = 0; i < n; i++)
    {
        cin >> arr[i];
    }

    sort(arr, arr+n);

    int start = 0;
    int end = 0;
    int count = 0;
    int sum = 0;

    while (1)
    {
        if (sum >= m)
            sum -= arr[start++];
        else if (end == n)
            break;
        else
            sum += arr[end++];
        if (sum == m)
            count++;
    }
    cout << count;
}