[백준] 2003 수들의 합 / 투포인터 (C++)

sobokii·2023년 10월 25일
0

문제풀이

목록 보기
32/52

문제
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)의 시간복잡도를 가질 것이다.
하지만 투포인터 알고리즘으로 하면 p1, p2가 왔다갔다하는 최대 횟수는 2N번이므로 시간복잡도가 O(N)이다.

#include <iostream>
#include <vector>
using namespace std;


int main()
{
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	
	int N, M;
	cin >> N >> M;
	vector<int> v(N);
	int p1 = 0, p2 = 0, sum = 0, cnt = 0;

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

	while (p1 < N && p2 < N)
	{
		while (p2 < N)
		{
			sum += v[p2];
			if (sum == M)
			{
				cnt++;
				p1++;
				p2 = p1;
				sum = 0;
				break;
			}
			else if (sum > M)
			{
				p1++;
				p2 = p1;
				sum = 0;
				break;
			}
			else
			{
				p2++;
			}
		}
	}

	cout << cnt;
	return 0;
}
profile
직장 구하고 있습니다.

0개의 댓글