취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 33026번 문제인 LOL Lovers이다.
문제는 아래 링크를 확인하자.

https://www.acmicpc.net/problem/33026

주어진 문자열을 둘로 나눌 수 있는 지점에서 모두 나눠보면서 그 두 문자열의 L과 O의 개수가 서로 다른지를 매번 확인해주자. \(N\)이 매우 작으므로 이와 같은 구현으로도 충분히 문제를 해결할 수 있다.

물론 전체 L과 O의 개수 및 각 접두사의 L과 O의 개수를 관리해 문제를 더 효율적으로 해결할 수도 있다. 아래의 구현은 이 풀이를 바탕으로 한다.

아래는 제출한 소스코드이다.

#include <iostream>
using namespace std;

int N, cntL, cntO, cntLL, cntOO;
string s;

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

	cin >> N >> s;
	for (auto &l:s) {
		if (l == 'L') cntL++;
		else cntO++;
	}
	for (int i = 0; i + 1 < N; i++) {
		if (s[i] == 'L') cntLL++;
		else cntOO++;
		if (cntLL != cntL - cntLL && cntOO != cntO - cntOO) {
			cout << i + 1;
			return 0;
		}
	}
	cout << -1;
}