[BOJ]2003 수들의 합2 C++

Min Kyu Jeon·2021년 7월 20일
1

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

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

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

일반적으로 완전탐색으로 풀이하게 되면 시간 복잡도가 O(N^2)이다.
2point 슬라이딩 윈도우를 이용하여 풀면 시간 복잡도는 O(2N)으로 줄일 수 있다.

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


int main() {
	ios::sync_with_stdio(0);
	cin.tie(0); cout.tie(0);

	int N, M;
	int arr[10001];
	cin >> N >> M;

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

	int l = 0;int r = 0;int sum = 0;int ans = 0;
	for (;;) {
		for (int i = l; i <= r; i++) {
			sum += arr[i];
		}
		if (sum == M) {
			ans++;
			l++;
			sum = 0;
		}
		else if (sum > M) {
			l++;
			sum = 0;
		}
		else {
			r++;
			sum = 0;
		}
		if (r >= N) {
			break;
		}
	}
	cout << ans;
	return 0;

}

0개의 댓글