[백준/c++] 14005번: 피카츄

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

 

문제

길이가 5000이 넘지 않는 알파벳 소문자로 이루어진 문자열 S를 입력받고, S가 "pi", "ka", "chu"로만 이루어져 있는 경우 "YES", 그렇지 않은 경우 "NO"를 출력한다.

 

풀이

문자열을 입력받고, 인덱스를 기준으로 2, 3개씩 문자열을 잘라서 비교하며 조건을 만족하는지 확인한다.

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

string s, ans = "YES";


int main() {

	cin >> s;

	int idx = 0;

	while (idx < s.length()) {
		if (s.substr(idx, 2) == "pi" || s.substr(idx, 2) == "ka") idx += 2;
		else if (s.substr(idx, 3) == "chu") idx += 3;
		else {
			ans = "NO";
			break;
		}
	}

	cout << ans;
}