[백준]2003_수들의 합 2

Solved

✔ n개의 수들과 어떤 수 m이 주어지고, 우리는 이 0<=i,j<=n-1 인덱스를 만족하는 i,j에 대해 arr[i]+...+arr[j]==m을 만족하는 경우의 수를 구해야 한다.
✔ N(1 ≤ N ≤ 10,000),M(1 ≤ M ≤ 300,000,000) 이다. 10000C500 을 계산하지 않더라도 모든 경우의 수를 계산한다면 시간 초과가 날 것임을 짐작할 수 있다.
특정 구간의 수들의 합이 M이 되는지 묻는 문제엔 투포인터 알고리즘이 적합하다.
✔ 투포인터는 배열에서 인덱스를 두개씩 다룬다. 예를 들어 만약 인덱스 i와 j 사이의 합이 m보다 부족하다면 오른쪽에 있는 인덱스를 +1 하고, m보다 크다면 좌측에 있는 인덱스를 +1하는 방식이다.

using namespace std;
#include <iostream>
#define MAX 10000

int input[MAX];
int main() {
	int n, m;
	cin >> n >> m;
	for (int i = 0; i < n; i++)
		cin >> input[i];

	int answer = 0, start = 0, end = 0;
	int sum = 0;
	while (end <= n) {
		if (sum >= m) sum -= input[start++];
		else if (sum < m) sum += input[end++];
		if (sum == m) answer++;
	}
	cout << answer << '\n';
}

좋은 웹페이지 즐겨찾기