[BOJ 백준] 수들의 합 2(2003) C++

 

링크 : https://www.acmicpc.net/problem/2003

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을 넘지 않는 자연수이다.

출력 : 

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

 

예제 입력 : 

10 5
1 2 3 4 2 5 3 1 1 2

예제 출력 : 

3

 

 

접근법 : 

1) 어떻게 풀 것인가?

힌트

 - 단순 완전탐색을 하면 N^2 = 약 1억의 시간복잡도로 time out 예상

 - "연속된 수의 합"

 

2) 시간복잡도

 

 

3) 공간복잡도

 

 

4) 풀면서 놓쳤던점

 

 

5) 이 문제를 통해 얻어갈 것

 

 

 

C++ 코드 : 

// 수들의 합 2 2003 C++
#if 1
#include <cstdio>
#pragma warning(disable:4996)

using namespace std;

int N, M;
int array[100000 + 3];

void input();

int main() {
	// 1. 입력받는다
	//freopen("input.txt", "r", stdin);
	input();

	// 2. 슬라이딩 윈도우
	// 초기화
	int low, high, ans, sum; 
	low = high = ans = sum = 0;

	// 1-1) sum < M, high 슬라이더 앞으로 (high < N일때)
	// 1-2) sum >= M, low 슬라이더 앞으로
	// 2) sum==M이면 추가
	while (1) {
		// 1-1) sum < M, high 슬라이더 앞으로 (high < N일때)
		if (sum < M) {
			// * 탈출조건 : high == N 이면 끝까지 왔으므로 탈출
			if (high == N) {
				break;
			}
			sum += array[high];
			high++;
		}

		// 1-2) sum >= M이면, low 슬라이더 앞으로
		else {  //(sum >= M) 
			sum -= array[low];
			low++;
		}

		// 2) 위 과정을 거치고 나서 sum == M이면 답에 추가
		if (sum == M) ans++;

	}

	// 3. 답을 출력한다.
	printf("%d", ans);

	return 0;
}

void input() {
	scanf("%d %d", &N, &M);
	// cin >> N>> M;
	for (int i = 0; i < N; i++) {
		scanf("%d", &array[i]);
	}
}
#endif