[프로그래머스/C++] 문자열 바꿔서 찾기

문제

문자 "A"와 "B"로 이루어진 문자열 myString과 pat가 주어집니다. myString의 "A"를 "B"로, "B"를 "A"로 바꾼 문자열의 연속하는 부분 문자열 중 pat이 있으면 1을 아니면 0을 return 하는 solution 함수를 완성하세요.

 

솔루션

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

int solution(string myString, string pat) {
    int answer = 0;
    string inverse = "";
    for(int i = 0; i < myString.size(); i++){
        if(myString[i] == 'A'){
            inverse += 'B';
        }
        else if(myString[i] == 'B'){
            inverse += 'A';
        }
    }
    if(inverse.find(pat) != string::npos){
        return 1;
    }
    else{
        return 0;
    }
}

 

https://school.programmers.co.kr/learn/courses/30/lessons/181864

프로그래머스