[프로그래머스] 옹알이 (2) c++

문제 설명

머쓱이는 태어난 지 11개월 된 조카를 돌보고 있습니다. 조카는 아직 "aya", "ye", "woo", "ma" 네 가지 발음과 네 가지 발음을 조합해서 만들 수 있는 발음밖에 하지 못하고 연속해서 같은 발음을 하는 것을 어려워합니다. 문자열 배열 babbling이 매개변수로 주어질 때, 머쓱이의 조카가 발음할 수 있는 단어의 개수를 return하도록 solution 함수를 완성해주세요.

제한 사항

• 1 ≤ babbling의 길이 ≤ 100
• 1 ≤ babbling[i]의 길이 ≤ 30
• 문자열은 알파벳 소문자로만 이루어져 있습니다.

입출력 예

babbling	result
["aya", "yee", "u", "maa"]	1
["ayaye", "uuu", "yeye", "yemawoo", "ayaayaa"]	2

풀이

#include <string>
#include <vector>

using namespace std;

int solution(vector<string> babbling) {
    int answer = 0;
    for(const auto& s : babbling){
        bool check = true;
        int flag = 0;
        for(int i = 0; i < s.length(); i++){
            if(s.substr(i, 3) == "aya" && flag != 1){
                flag = 1;
                i += 2;
            }
            else if(s.substr(i, 2) == "ye" && flag != 2){
                flag = 2;
                i += 1;
            }
            else if(s.substr(i, 3) == "woo" && flag != 3){
                flag = 3;
                i += 2;
            }
            else if(s.substr(i, 2) == "ma" && flag != 4){
                flag = 4;
                i += 1;
            }
            else{
                check = false;
                break;
            }
        }
        if(check){
            answer++;
        }
    }
    return answer;
}