5 - B 질문입니다! - 인프런 | 커뮤니티 질문&답변

10주완성 C++ 코딩테스트 | 알고리즘 코딩테스트

5-B : erase()를 이용한 풀이

5 - B 질문입니다!

작성

113

#include <bits/stdc++.h>

using namespace std;

string str, fire_str;
string ret="";

int main()
{
    cin >> str >> fire_str;
    for (char i:str){
        ret+=i;
        if (i==fire_str[fire_str.size()-1]){
            if (ret.size()-fire_str.size()<0) continue;
            else if (ret.substr(ret.size()-fire_str.size(),fire_str.size())==fire_str){
                ret.erase(ret.size()-fire_str.size(),fire_str.size());
            }
        }
    }
    if (ret=="") cout << "FRULA";
    else cout << ret;
}

out of range 제출하게 되면 out of range 에러가 뜨는데 이유를 모르겠습니다. 어디서 나는 건가요??

c++코딩-테스트

답변 2

큰돌

지식공유자

안녕하세요 JOXX님 ㅎㅎ

코드 중에

fire_str.size()-1

이부분에서 size가 0이면 -1를 참조하기 때문에 해당 에러가 발생하게 됩니다.

감사합니다.

JOXXEP

질문자

답변 감사합니다...! 앞으로는 더욱 신경써야겠습니다!

인프런 AI 인턴

안녕하세요, 인프런 AI 인턴입니다.

제출하신 C++ 코드에서 out of range 에러가 발생하는 부분은 다음 조건문에서 발생할 수 있습니다.

if (i==fire_str[fire_str.size()-1]){
    if (ret.size()-fire_str.size()<0) continue;
    else if (ret.substr(ret.size()-fire_str.size(),fire_str.size())==fire_str){
        ret.erase(ret.size()-fire_str.size(),fire_str.size());
    }
}

여기서 fire_str[fire_str.size()-1] 부분에서 fire_str.size()가 0일 때 fire_str[fire_str.size()-1]를 실행하면 fire_str[-1]과 같은 상황이 발생해 out of range 에러가 나타납니다. fire_str이 빈 문자열인 경우를 대비해야 합니다.

이를 해결하기 위해서는 fire_str 의 사이즈가 0인지 먼저 확인하는 조건문을 추가하는 것이 좋습니다. 또한, ret.substr에서 ret.size() - fire_str.size()가 음수가 되지 않도록 하는 검사도 잘 하고 계시니, fire_str.size() > 0임을 확인하는 조건을 추가하면 out of range 에러를 방지할 수 있습니다.

예를 들어, 다음과 같이 조건을 추가할 수 있습니다.

if (!fire_str.empty() && i == fire_str[fire_str.size() - 1]) {
    if (ret.size() - fire_str.size() < 0) continue;
    else if (ret.substr(ret.size() - fire_str.size(), fire_str.size()) == fire_str) {
        ret.erase(ret.size() - fire_str.size(), fire_str.size());
    }
}

이렇게 fire_str이 빈 문자열이 아닐 때만 해당 로직을 실행하는 조건을 추가하면 out of range 에러를 방지할 수 있습니다.

인프런 커뮤니티 질문&답변

5 - B 질문입니다!