[백준 9996] 한국이 그리울 땐 서버에 접속하지

문제

9996번: 한국이 그리울 땐 서버에 접속하지

풀이

실패한 코드

로그인

아이디어

1. 주어진 패턴에서 * 가 위치한 인덱스 번호를 찾는다.
2. 입력한 파일명의 prefix 와 suffix 를 추출한다.
3. 패턴과 대조한 결과를 출력한다.

문제점

1. substr() 를 사용하지 않고 무분별한 반복문 사용한 것
2. find() 를 사용하지 않고 반복문을 사용해 * 의 위치를 찾은 것
3. 파일명의 길이가 패턴의 prefix 와 suffix 의 길이 합보다 작은 경우를 파악하지 못한 것

#include <bits/stdc++.h>
using namespace std;
int cnt, idx;
string pattern;
string startPattern, endPattern;
void find() {
    for (int i = 0; i < 100; i++) {
        if (pattern[i] == '*') {
            idx = i;
            return;
        }
    }
}

int main() {
    cin >> cnt;
    cin >> pattern;
    find();
    for (int i = 0; i < idx; i++) startPattern += pattern[i];
    for (int i = idx + 1; i < pattern.length(); i++) endPattern += pattern[i];
    for (int i = 0; i < cnt; i++) {
        string input;
        cin >> input;
        string start;
        for (int j = 0; j < idx; j++) start += input[j];
        string end;
        for (int k = input.length() - idx; k < input.length(); k++) end += input[k];
        if ((start == startPattern) && (end == endPattern)) {
            cout << "DA" << "\n";
        } else {
            cout << "NE" << "\n";
        }
    }
}

해결 코드

로그인

아이디어

1. ab*ba 패턴일 때 파일명이 aba 인 경우
   1. 즉, 패턴의 prefix / suffix 의 길이 합이 파일명보다 긴 경우 NE 출력
2. 불필요한 반복문은 제거하고 find(), substr() 함수 사용

#include <bits/stdc++.h>
using namespace std;
int cnt;
string pattern, input;
int main() {
    cin >> cnt;
    cin >> pattern;
    int pos = pattern.find('*');
    string prefix = pattern.substr(0, pos);
    string suffix = pattern.substr(pos + 1);
    for (int i = 0; i < cnt; i++) {
        cin >> input;
        if (prefix.size() + suffix.size() > input.size()) cout << "NE\n";
        else {
            if (input.substr(0, pos) == prefix && input.substr(input.size() - suffix.length()) == suffix)cout << "DA\n";
            else cout << "NE\n";
        }
    }
}