본문 바로가기

백준/골드

[백준 16120번] PPAP (C++)

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

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

string s;
int cnt = 0;

int main()
{
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	cin >> s;
	for (int i = 0; i < s.length(); i++)
	{
		if (s[i] == 'P') { cnt++; } //"P"일때 카운트 증가
		else if (cnt >= 2 && s[i + 1] == 'P') { cnt -= 2; }  //"PPAP"일때, 카운트 -=2
		else { cout << "NP"; return 0; }  //입력값이 'PPAPA'인 경우 cnt == 1이지만 PPAP가 아님
	}
	if (cnt == 1) { cout << "PPAP"; } //맞으면 "PPAP" 출력
	else { cout << "NP"; }            //아니면 "NP" 출력
}

"P"일때는 카운트를 증가시켰다가, 앞에 카운트가 2이상(PP)일때, 뒤에가 P일때 즉 "PPAP"일때는 카운트를 2감소시켰다.

(PPPAP기준 최종 카운트는 2가됨[PPPAP -> PP])

이를 통한 카운트 최종값이 1일때는 "PPAP"이다.

하지만 for문에서 else로 걸러주지 않는다면 "PPAPA"인 경우 카운트가 1이기에 PPAP로 출력하게되는 오류가 발생한다.